[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