better MaterializeMergeSchedule.cpp with %lane indexed batch computes

support for tensors of index values
2026-05-22 21:52:28 +02:00
parent 495186503c
commit c77ffa9c56
20 changed files with 398 additions and 300 deletions
@@ -427,11 +427,7 @@ LogicalResult GemvToSpatialCompute::matchAndRewrite(ONNXGemmOp gemmOp,
        auto inputArg = computeOp.getInputArgument(aHSliceId);
        if (!weightArg || !inputArg)
          return failure();
-        vmmOutputs.push_back(spatial::SpatVMMOp::create(rewriter,
-                                                        gemmLoc,
-                                                        currOutHSliceType,
-                                                        *weightArg,
-                                                        *inputArg));
+        vmmOutputs.push_back(spatial::SpatVMMOp::create(rewriter, gemmLoc, currOutHSliceType, *weightArg, *inputArg));
      }
      if (vmmOutputs.empty()) {
        gemmOp.emitOpError("requires at least one non-empty slice when lowering tiled Gemm to Spatial VMMs");