diff --git a/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp b/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp index e58503d508ce..ab4d90ec745d 100644 --- a/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp +++ b/mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp @@ -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(createLinalgGeneralizeNamedOpsPass()); + pm.addNestedPass(createLinalgElementwiseOpFusionPass()); // Sparsification and bufferization mini-pipeline. pm.addPass(createSparsificationAndBufferizationPass(