chore: return self in aten::extend

ruoqianguo · narendasan · commit e1b68fc0ae15 · 2022-02-23T18:59:28.000-08:00
Signed-off-by: Ruoqian Guo &lt;ruoqiang@nvidia.com&gt;
diff --git a/core/conversion/conversion.cpp b/core/conversion/conversion.cpp
@@ -416,7 +416,7 @@ void ConvertBlockToNetDef(
       EvaluateConditionalBlock(ctx, n);
     } else if (to_eval) {
       auto eval = EvaluateNode(ctx, n);
-      if (eval) {
+      if (eval && n->outputs().size() > 0) {
         if (n->outputs().size() > 1) { // For ListUnpack scenario
           if (eval.value().isTuple()) {
             auto eval_list = eval.value().toTuple();
diff --git a/core/conversion/evaluators/aten.cpp b/core/conversion/evaluators/aten.cpp
@@ -295,6 +295,7 @@ auto aten_registrations TORCHTRT_UNUSED =
                         for (int64_t i = 0; i < other_size; i++) {
                           self.push_back(other.get(i));
                         }
+                        return self;
                       } else {
                         TORCHTRT_THROW_ERROR(
                             "Unimplemented data type for aten::extend.t evaluator: "

Original file line number	Diff line number	Diff line change
`@@ -295,6 +295,7 @@ auto aten_registrations TORCHTRT_UNUSED =`
`295`	`295`	`for (int64_t i = 0; i < other_size; i++) {`
`296`	`296`	`self.push_back(other.get(i));`
`297`	`297`	`}`
	`298`	`+ return self;`
`298`	`299`	`} else {`
`299`	`300`	`TORCHTRT_THROW_ERROR(`
`300`	`301`	`"Unimplemented data type for aten::extend.t evaluator: "`