[Mlir-commits] [mlir] 910d627 - [mlir][sparse] avoid nop rewriting on runtime lib path in pipeline
Aart Bik
llvmlistbot at llvm.org
Mon Nov 14 16:39:01 PST 2022
Author: Aart Bik
Date: 2022-11-14T16:38:47-08:00
New Revision: 910d6270dba7f1195946d5e3e74265d736839235
URL: https://github.com/llvm/llvm-project/commit/910d6270dba7f1195946d5e3e74265d736839235
DIFF: https://github.com/llvm/llvm-project/commit/910d6270dba7f1195946d5e3e74265d736839235.diff
LOG: [mlir][sparse] avoid nop rewriting on runtime lib path in pipeline
Reviewed By: Peiming
Differential Revision: https://reviews.llvm.org/D137981
Added:
Modified:
mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp
Removed:
################################################################################
diff --git a/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp b/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp
index 0e1fbafc5f760..e48a760df0fa2 100644
--- a/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp
+++ b/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp
@@ -59,13 +59,15 @@ void mlir::sparse_tensor::buildSparseCompiler(
return;
pm.addPass(createSparseTensorRewritePass(options.enableRuntimeLibrary));
pm.addPass(createSparsificationPass(options.sparsificationOptions()));
- if (options.enableRuntimeLibrary)
+ if (options.enableRuntimeLibrary) {
pm.addPass(createSparseTensorConversionPass(
options.sparseTensorConversionOptions()));
- else
+ } else {
pm.addPass(
createSparseTensorCodegenPass(options.enableBufferInitialization));
- pm.addPass(createSparseBufferRewritePass(options.enableBufferInitialization));
+ pm.addPass(
+ createSparseBufferRewritePass(options.enableBufferInitialization));
+ }
pm.addPass(createDenseBufferizationPass(
getBufferizationOptions(/*analysisOnly=*/false)));
pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
More information about the Mlir-commits
mailing list