[Mlir-commits] [mlir] [mlir] Fix bug in UnPackOp tiling implementation causing infinite loop (PR #113571)

llvmlistbot at llvm.org llvmlistbot at llvm.org
Thu Oct 24 09:31:49 PDT 2024


================
@@ -554,13 +554,14 @@ struct UnPackOpTiling
     sliceSrcIndices.append(numInnerTiles, zeroAttr);
     sliceSrcSizes.append(unpackOp.getMixedTiles());
     sliceSrcStrides.append(numInnerTiles, oneAttr);
-    Value sliceSource =
+    SmallVector<Operation *> generatedSlices;
+    ExtractSliceOp sliceSource =
         b.create<ExtractSliceOp>(loc, unpackOp.getSource(), sliceSrcIndices,
                                  sliceSrcSizes, sliceSrcStrides);
+    generatedSlices.push_back(sliceSource);
----------------
Max191 wrote:

I'm realizing that maybe I'm not supposed to add the input operand slices. Should I be doing this?

https://github.com/llvm/llvm-project/pull/113571


More information about the Mlir-commits mailing list