[Mlir-commits] [mlir] [mlir][vector] Add support for multi-dim reduction vector distribution (PR #71193)

Lei Zhang llvmlistbot at llvm.org
Thu Nov 9 22:48:01 PST 2023


================
@@ -829,20 +831,24 @@ struct WarpOpTransferRead : public OpRewritePattern<WarpExecuteOnLane0Op> {
                            warpOp.getLaneid(), delinearizedIds))
       return rewriter.notifyMatchFailure(
           read, "cannot delinearize lane ID for distribution");
-    assert(!delinearizedIds.empty() || map.getNumResults() == 0);
 
-    for (auto it : llvm::zip_equal(indexMap.getResults(), map.getResults())) {
+    SmallVector<Value, 4> indices(read.getIndices().begin(),
+                                  read.getIndices().end());
+    indexMap.dump();
----------------
antiagainst wrote:

Leftover debugging prints?

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


More information about the Mlir-commits mailing list