Skip to content

Commit 699012c

Browse files
inocsinnarendasan
authored andcommitted
update test_remove_detach_pass
Signed-off-by: inocsin <[email protected]>
1 parent 374aafd commit 699012c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/core/lowering/test_remove_detach_pass.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,16 +10,16 @@ TEST(LoweringPasses, RemoveDetachCorrectly) {
1010
std::string source_graph = R"IR(
1111
graph(%input):
1212
%2 = aten::detach(%input)
13-
%3 = foo::bar(%2)
13+
%3 = aten::sin(%2)
1414
return (%3))IR";
1515
std::string target_graph = R"IR(
1616
graph(%input):
17-
%3 = foo::bar(%input)
17+
%3 = aten::sin(%input)
1818
return (%3))IR";
1919

2020
auto sg = std::make_shared<torch::jit::Graph>();
2121
torch::jit::parseIR(source_graph, &*sg);
22-
trtorch::core::lowering::passes::RemoveContiguous(sg);
22+
trtorch::core::lowering::passes::RemoveTo(sg);
2323

2424
auto tg = std::make_shared<torch::jit::Graph>();
2525
torch::jit::parseIR(target_graph, &*tg);

0 commit comments

Comments
 (0)