[Mlir-commits] [mlir] [mlir][sparse] end-to-end matmul between Dense and BSR tensors (PR #71448)
Peiming Liu
llvmlistbot at llvm.org
Tue Nov 7 09:55:19 PST 2023
================
@@ -139,13 +139,13 @@ class SparsificationAndBufferizationPass
// of `bufferization.alloc_tensor` ops.
{
OpPassManager pm("builtin.module");
- pm.addPass(
- createSparseReinterpretMapPass(ReinterpretMapScope::kGenericOnly));
+ // We need to reinterpret maps on GenericOp, EmptyOp, and AllocTensorOp.
+ // empty BSR
+ pm.addPass(createSparseReinterpretMapPass(ReinterpretMapScope::kAll));
pm.addPass(createSparsificationPass(sparsificationOptions));
pm.addNestedPass<func::FuncOp>(createStageSparseOperationsPass());
----------------
PeimingLiu wrote:
Yeah, GPU libgen needs the rich type information.
https://github.com/llvm/llvm-project/pull/71448
More information about the Mlir-commits
mailing list