Fix some testing

Author: josel-amd

mlir/lib/Dialect/Linalg/Utils/Utils.cpp

@@ -798,9 +798,15 @@ computeAllSliceParameters(OpBuilder &builder, Location loc, LinalgOp linalgOp,
     // transformations such as padding and bufferization since the
     // extract/insert slice pairs make the accessed iteration argument
     // subdomains explicit.
-
+    SmallVector<int64_t> domainSizes;
+    // FIXME: tileToPartialReduction adds the new init tensor to the output
+    // but doesn't update the indexing type of the index map causing a crash.
+    // isAllParallelLoops
+    if (linalgOp.getNumParallelLoops() == linalgOp.getNumLoops()) {
+      domainSizes = linalgOp.getStaticLoopRanges();
+    }
     Type operandType = opOperand.get().getType();
-    if (!isTiled(map, tileSizes, linalgOp.getStaticLoopRanges()) &&
+    if (!isTiled(map, tileSizes, domainSizes) &&
         !(isa<RankedTensorType>(operandType) &&
           linalgOp.isDpsInit(&opOperand))) {
       allSliceParams.push_back(std::nullopt);
@@ -812,7 +818,7 @@ computeAllSliceParameters(OpBuilder &builder, Location loc, LinalgOp linalgOp,
 
     allSliceParams.push_back(computeSliceParameters(
         builder, loc, shapedOp, tileSizes, map, lbs, sizeBounds, subShapeSizes,
-        omitPartialTileCheck, linalgOp.getStaticLoopRanges()));
+        omitPartialTileCheck, domainSizes));
   }
 
   return allSliceParams;

mlir/test/Dialect/Linalg/tile-tensors.mlir

@@ -170,6 +170,10 @@ module attributes {transform.with_named_sequence} {
 
 // -----
 
+
+// CHECK-LABEL: func @generic_op_tensors
+//  CHECK-SAME:   %[[ARG0:[a-zA-Z0-9_]+]]: tensor<?x?x?xf32>
+//  CHECK-SAME:   %[[ARG1:[a-zA-Z0-9_]+]]: tensor<?x?x?xf32>
 func.func @non_monotonic_affine_expr(%arg0 : tensor<7xf32>) -> tensor<7xf32> {
   %c0 = arith.constant 0 : index
   %0 = tensor.dim %arg0, %c0 : tensor<7xf32>