fix comment

llvm · Aug 16, 2024 · d7fe9c7 · d7fe9c7
1 parent 89ef377
commit d7fe9c7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp b/mlir/lib/Dialect/Tensor/IR/TensorTilingInterfaceImpl.cpp
@@ -255,6 +255,9 @@ struct PackOpTiling
       ArrayRef<OpFoldResult> offsets, ArrayRef<OpFoldResult> sizes,
       SmallVectorImpl<OpFoldResult> &resultOffsets,
       SmallVectorImpl<OpFoldResult> &resultSizes) const {
+    if (operandNumber != 0)
+      return failure();
+
     auto packOp = cast<PackOp>(op);
     Location loc = packOp.getLoc();
 
@@ -299,6 +302,9 @@ struct PackOpTiling
   FailureOr<TilingResult> getTiledImplementationFromOperandTile(
       Operation *op, OpBuilder &b, unsigned operandNumber,
       ArrayRef<OpFoldResult> offsets, ArrayRef<OpFoldResult> sizes) const {
+    if (operandNumber != 0)
+      return failure();
+
     auto packOp = cast<PackOp>(op);
     Location loc = packOp.getLoc();