intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-26 21:53:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mlir][sparse] add linalg elt-wise fusion to sparsifier pipeline (#90924)

Browse Source 
yields better IR in general, and all end-to-end tests pass!
This commit is contained in:
Aart Bik
2024-05-02 18:21:00 -07:00
committed by GitHub
parent 6d7d8e5b59
commit 1949856126
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp
View File

@@ -31,8 +31,9 @@
void mlir::sparse_tensor::buildSparsifier(OpPassManager &pm,
const SparsifierOptions &options) {
// Rewrite named linalg ops into generic ops.
// Rewrite named linalg ops into generic ops and apply fusion.
pm.addNestedPass<func::FuncOp>(createLinalgGeneralizeNamedOpsPass());
pm.addNestedPass<func::FuncOp>(createLinalgElementwiseOpFusionPass());
// Sparsification and bufferization mini-pipeline.
pm.addPass(createSparsificationAndBufferizationPass(