[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