fix(remove_to): bug in remove_to.cpp, replace outputs()[0] with input…

…s()[0] Signed-off-by: inocsin <[email protected]>
pytorch · Jan 28, 2021 · 6c5118a · 6c5118a
1 parent 885439c
commit 6c5118a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/lowering/passes/remove_to.cpp b/core/lowering/passes/remove_to.cpp
@@ -33,7 +33,7 @@ struct ToRemoval {
       auto n = *it;
       if (n->kind() == c10::Symbol::fromQualString("aten::to")) {
         LOG_GRAPH("Found that node " << *n << "  is an to node (RemoveTo)" << std::endl);
-        n->outputs()[0]->replaceAllUsesWith(n->inputs()[1]);
+        n->outputs()[0]->replaceAllUsesWith(n->inputs()[0]);
         it.destroyCurrent();
       }
     }