@@ -524,8 +524,8 @@ TEST(Evaluators, EqStrResultIsTrueEvaluatesCorrectly) {
524524 auto g = std::make_shared<torch::jit::Graph>();
525525 torch::jit::parseIR (graph, g.get ());
526526
527- auto jit_results = trtorch ::tests::util::EvaluateGraphJIT (g, {});
528- auto trt_results = trtorch ::tests::util::EvaluateGraph (g->block (), {});
527+ auto jit_results = torch_tensorrt ::tests::util::EvaluateGraphJIT (g, {});
528+ auto trt_results = torch_tensorrt ::tests::util::EvaluateGraph (g->block (), {});
529529
530530 ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
531531}
@@ -541,8 +541,8 @@ TEST(Evaluators, EqStrResultIsFalseEvaluatesCorrectly) {
541541 auto g = std::make_shared<torch::jit::Graph>();
542542 torch::jit::parseIR (graph, g.get ());
543543
544- auto jit_results = trtorch ::tests::util::EvaluateGraphJIT (g, {});
545- auto trt_results = trtorch ::tests::util::EvaluateGraph (g->block (), {});
544+ auto jit_results = torch_tensorrt ::tests::util::EvaluateGraphJIT (g, {});
545+ auto trt_results = torch_tensorrt ::tests::util::EvaluateGraph (g->block (), {});
546546
547547 ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
548548}
@@ -558,8 +558,8 @@ TEST(Evaluators, AndBoolResultIsTrueEvaluatesCorrectly) {
558558 auto g = std::make_shared<torch::jit::Graph>();
559559 torch::jit::parseIR (graph, g.get ());
560560
561- auto jit_results = trtorch ::tests::util::EvaluateGraphJIT (g, {});
562- auto trt_results = trtorch ::tests::util::EvaluateGraph (g->block (), {});
561+ auto jit_results = torch_tensorrt ::tests::util::EvaluateGraphJIT (g, {});
562+ auto trt_results = torch_tensorrt ::tests::util::EvaluateGraph (g->block (), {});
563563
564564 ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
565565}
@@ -575,8 +575,8 @@ TEST(Evaluators, AndBoolResultIsFalseEvaluatesCorrectly) {
575575 auto g = std::make_shared<torch::jit::Graph>();
576576 torch::jit::parseIR (graph, g.get ());
577577
578- auto jit_results = trtorch ::tests::util::EvaluateGraphJIT (g, {});
579- auto trt_results = trtorch ::tests::util::EvaluateGraph (g->block (), {});
578+ auto jit_results = torch_tensorrt ::tests::util::EvaluateGraphJIT (g, {});
579+ auto trt_results = torch_tensorrt ::tests::util::EvaluateGraph (g->block (), {});
580580
581581 ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
582582}
0 commit comments