@@ -524,8 +524,8 @@ TEST(Evaluators, EqStrResultIsTrueEvaluatesCorrectly) {
524
524
auto g = std::make_shared<torch::jit::Graph>();
525
525
torch::jit::parseIR (graph, g.get ());
526
526
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 (), {});
529
529
530
530
ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
531
531
}
@@ -541,8 +541,8 @@ TEST(Evaluators, EqStrResultIsFalseEvaluatesCorrectly) {
541
541
auto g = std::make_shared<torch::jit::Graph>();
542
542
torch::jit::parseIR (graph, g.get ());
543
543
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 (), {});
546
546
547
547
ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
548
548
}
@@ -558,8 +558,8 @@ TEST(Evaluators, AndBoolResultIsTrueEvaluatesCorrectly) {
558
558
auto g = std::make_shared<torch::jit::Graph>();
559
559
torch::jit::parseIR (graph, g.get ());
560
560
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 (), {});
563
563
564
564
ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
565
565
}
@@ -575,8 +575,8 @@ TEST(Evaluators, AndBoolResultIsFalseEvaluatesCorrectly) {
575
575
auto g = std::make_shared<torch::jit::Graph>();
576
576
torch::jit::parseIR (graph, g.get ());
577
577
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 (), {});
580
580
581
581
ASSERT_TRUE (jit_results[0 ] == trt_results[0 ]);
582
582
}
0 commit comments