RewritePatternSet &patterns,
bool enableBufferInitialization);
+std::unique_ptr<Pass> createSparseTensorCodegenPass();
std::unique_ptr<Pass>
-createSparseTensorCodegenPass(bool enableBufferInitialization = false);
+createSparseTensorCodegenPass(bool enableBufferInitialization);
//===----------------------------------------------------------------------===//
// The SparseTensorRewriting pass.
void populateSparseBufferRewriting(RewritePatternSet &patterns,
bool enableBufferInitialization);
+
+std::unique_ptr<Pass> createSparseBufferRewritePass();
std::unique_ptr<Pass>
-createSparseBufferRewritePass(bool enableBufferInitialization = false);
+createSparseBufferRewritePass(bool enableBufferInitialization);
//===----------------------------------------------------------------------===//
// Registration.
return std::make_unique<SparseTensorConversionPass>(options);
}
+std::unique_ptr<Pass> mlir::createSparseTensorCodegenPass() {
+ return std::make_unique<SparseTensorCodegenPass>();
+}
+
std::unique_ptr<Pass>
mlir::createSparseTensorCodegenPass(bool enableBufferInitialization) {
return std::make_unique<SparseTensorCodegenPass>(enableBufferInitialization);
}
+std::unique_ptr<Pass> mlir::createSparseBufferRewritePass() {
+ return std::make_unique<SparseBufferRewritePass>();
+}
+
std::unique_ptr<Pass>
mlir::createSparseBufferRewritePass(bool enableBufferInitialization) {
return std::make_unique<SparseBufferRewritePass>(enableBufferInitialization);