Skip to content

Commit ba743f5

Browse files
committed
fix(eval): Rollback 1.11a0 change + namespace issues
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 540e135 commit ba743f5

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

core/conversion/evaluators/eval_macros.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@
6363
auto b = args.at(n->input(1)).unwrapToString(); \
6464
return operation; \
6565
} else { \
66-
TRTORCH_THROW_ERROR( \
66+
TORCHTRT_THROW_ERROR( \
6767
"Unimplemented data type for " \
6868
<< node_kind << " evaluator b arg:" << args.at(n->input(1)).IValue()->type()->str()); \
6969
return {}; \

core/conversion/evaluators/eval_util.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ at::Tensor createTensorFromList(
238238
/// Gets shape of tensor to be created
239239
auto sizes = compute_sizes(data);
240240
checkListInputType(elem_type, sizes.size() == 1 && sizes[0] == 0);
241-
at::ScalarType initial_scalar_type = c10::scalarTypeFromJitType(*elem_type);
241+
at::ScalarType initial_scalar_type = c10::scalarTypeFromJitType(elem_type);
242242
if (initial_scalar_type == at::ScalarType::Double) {
243243
initial_scalar_type = at::typeMetaToScalarType(c10::get_default_dtype());
244244
}

tests/core/conversion/evaluators/test_aten_evaluators.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)