Skip to content

Commit 0bf1231

Browse files
authored
fix: change floordiv to divmod for //tests/core/lowering:test_remove_unnecessary_casts (#3223)
1 parent f3636a9 commit 0bf1231

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/lowering/passes/remove_unnecessary_casts.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ void RemoveSingleUse0DTensors(std::shared_ptr<torch::jit::Graph>& g) {
129129
new_node->outputs()[0]->setType(c10::IntType::get());
130130
user->outputs()[0]->replaceAllUsesWith(new_node->outputs()[0]);
131131
user->destroy();
132-
} else if (user->kind() == c10::Symbol::fromQualString("aten::floordiv")) {
132+
} else if (user->kind() == c10::Symbol::fromQualString("aten::floor_divide")) {
133133
new_node = g->create(c10::aten::floordiv, user->inputs(), 1);
134134
new_node->insertAfter(user);
135135
new_node->outputs()[0]->setType(c10::IntType::get());

0 commit comments

Comments
 (0)