update RunEngineDynamic for allowshapetensors

mfeliz-cruise · mfeliz-cruise · commit b236435bf5b7 · 2023-08-02T13:26:15.000-07:00
diff --git a/tests/core/conversion/evaluators/test_aten_evaluators.cpp b/tests/core/conversion/evaluators/test_aten_evaluators.cpp
@@ -336,7 +336,7 @@ TEST(Evaluators, ZerosLikeDynamic) {
 
   auto jit_results = torch_tensorrt::tests::util::EvaluateGraphJIT(g, {in});
   auto params = torch_tensorrt::core::ir::get_static_params(g->inputs(), {});
-  auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in});
+  auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in}, true, true);
 
   ASSERT_TRUE(at::equal(jit_results[0].toTensor().to(at::kCUDA), trt_results[0]));
 }
@@ -392,7 +392,7 @@ TEST(Evaluators, OnesLikeDynamic) {
 
   auto jit_results = torch_tensorrt::tests::util::EvaluateGraphJIT(g, {in});
   auto params = torch_tensorrt::core::ir::get_static_params(g->inputs(), {});
-  auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in});
+  auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in}, true, true);
 
   ASSERT_TRUE(at::equal(jit_results[0].toTensor().to(at::kCUDA), trt_results[0]));
 }

Original file line number	Diff line number	Diff line change
`@@ -336,7 +336,7 @@ TEST(Evaluators, ZerosLikeDynamic) {`
`336`	`336`
`337`	`337`	`auto jit_results = torch_tensorrt::tests::util::EvaluateGraphJIT(g, {in});`
`338`	`338`	`auto params = torch_tensorrt::core::ir::get_static_params(g->inputs(), {});`
`339`		`- auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in});`
	`339`	`+ auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in}, true, true);`
`340`	`340`
`341`	`341`	`ASSERT_TRUE(at::equal(jit_results[0].toTensor().to(at::kCUDA), trt_results[0]));`
`342`	`342`	`}`
`@@ -392,7 +392,7 @@ TEST(Evaluators, OnesLikeDynamic) {`
`392`	`392`
`393`	`393`	`auto jit_results = torch_tensorrt::tests::util::EvaluateGraphJIT(g, {in});`
`394`	`394`	`auto params = torch_tensorrt::core::ir::get_static_params(g->inputs(), {});`
`395`		`- auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in});`
	`395`	`+ auto trt_results = torch_tensorrt::tests::util::RunGraphEngineDynamic(g, params, {in}, true, true);`
`396`	`396`
`397`	`397`	`ASSERT_TRUE(at::equal(jit_results[0].toTensor().to(at::kCUDA), trt_results[0]));`
`398`	`398`	`}`