update

Author: yyp0

lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp

@@ -7368,10 +7368,19 @@ class DecomposeAtenAdaptiveMaxPool1dOp
         loc, Torch::ListType::get(Torch::IntType::get(context)),
         ValueRange{constantOne});
 
-    rewriter.replaceOpWithNewOp<AtenMaxPool1dWithIndicesOp>(
-        op, op.getType(0), op.getType(1), input, kernelSizeList, strideList,
-        paddingSizeList, dialationList,
-        /*ceil_mode=*/constantFalse);
+    if (op.getResult(1).use_empty()) {
+      auto maxPool = rewriter.create<AtenMaxPool1dOp>(loc, op.getType(0), input, kernelSizeList,
+                                      strideList, paddingSizeList,
+                                      dialationList,
+                                      /*ceil_mode=*/constantFalse);
+      rewriter.replaceOp(op, {maxPool.getResult(), Value()});
+    } else {
+      auto maxPool = rewriter.create<AtenMaxPool1dWithIndicesOp>(loc, op.getType(0), op.getType(1), input, kernelSizeList,
+                                      strideList, paddingSizeList,
+                                      dialationList,
+                                      /*ceil_mode=*/constantFalse);
+      rewriter.replaceOp(op, {maxPool.getResult(0), maxPool.getResult(1)});
+    }
     return success();
   }
 };