fix(aten::neg): Fix a index bug in neg

Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · Jun 5, 2020 · 1b2cde4 · 1b2cde4
1 parent 2b23874
commit 1b2cde4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/conversion/evaluators/aten.cpp b/core/conversion/evaluators/aten.cpp
@@ -165,7 +165,7 @@ auto aten_registrations = RegisterNodeEvaluators()
     }).evaluator({
         c10::Symbol::fromQualString("aten::neg"),
         [](const torch::jit::Node* n, kwargs& args) -> c10::optional<torch::jit::IValue> {
-            auto el = args.at(n->input(1)).unwrapToInt();
+            auto el = args.at(n->input(0)).unwrapToInt();
 
             return el * -1;
         },