[Mlir-commits] [mlir] a0568ea - [mlir][sparse] Add dependence on bufferization.

[llvmlistbot at llvm.org]

Author: bixia1
Date: 2022-12-07T15:18:36-08:00
New Revision: a0568eabaf85cdb4f89bda9e75a49a70c165af7c

URL: https://github.com/llvm/llvm-project/commit/a0568eabaf85cdb4f89bda9e75a49a70c165af7c
DIFF: https://github.com/llvm/llvm-project/commit/a0568eabaf85cdb4f89bda9e75a49a70c165af7c.diff

LOG: [mlir][sparse] Add dependence on bufferization.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D139571

Added: 
    

Modified: 
    mlir/lib/Dialect/SparseTensor/Transforms/SparsificationAndBufferizationPass.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparsificationAndBufferizationPass.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparsificationAndBufferizationPass.cpp
index 6066a17e27914..5633714ed807e 100644
--- a/mlir/lib/Dialect/SparseTensor/Transforms/SparsificationAndBufferizationPass.cpp
+++ b/mlir/lib/Dialect/SparseTensor/Transforms/SparsificationAndBufferizationPass.cpp
@@ -9,6 +9,7 @@
 #include "mlir/Dialect/SparseTensor/Transforms/Passes.h"
 
 #include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"
+#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
 #include "mlir/Dialect/Bufferization/Transforms/Bufferize.h"
 #include "mlir/Dialect/Bufferization/Transforms/OneShotAnalysis.h"
 #include "mlir/Dialect/Bufferization/Transforms/OneShotModuleBufferize.h"
@@ -86,6 +87,10 @@ class SparsificationAndBufferizationPass
     return success();
   }
 
+  void getDependentDialects(::mlir::DialectRegistry &registry) const override {
+    registry.insert<bufferization::BufferizationDialect>();
+  }
+
   void runOnOperation() override {
     {
       // Run enabling transformations.