19 #ifndef MLIR_DIALECT_SPARSETENSOR_TRANSFORMS_PASSES_H_ 20 #define MLIR_DIALECT_SPARSETENSOR_TRANSFORMS_PASSES_H_ 26 namespace bufferization {
27 struct OneShotBufferizationOptions;
72 : parallelizationStrategy(p), vectorizationStrategy(v), vectorLength(vl),
73 enableSIMDIndex32(e), enableVLAVectorization(vla) {}
124 : sparseToSparseStrategy(s2s) {}
138 std::unique_ptr<Pass>
155 #define GEN_PASS_REGISTRATION 156 #include "mlir/Dialect/SparseTensor/Transforms/Passes.h.inc" 160 #endif // MLIR_DIALECT_SPARSETENSOR_TRANSFORMS_PASSES_H_
Include the generated interface declarations.
SparseTensorConversionOptions()
SparseToSparseConversionStrategy sparseToSparseStrategy
SparseParallelizationStrategy
Defines a parallelization strategy.
SparseToSparseConversionStrategy
Defines a strategy for implementing sparse-to-sparse conversion.
std::unique_ptr< Pass > createDenseBufferizationPass(const bufferization::OneShotBufferizationOptions &options)
SparseToSparseConversionStrategy sparseToSparseConversionStrategy(int32_t flag)
Converts command-line sparse2sparse flag to the strategy enum.
bool enableVLAVectorization
SparseTensorConversion options.
void populateSparseTensorConversionPatterns(TypeConverter &typeConverter, RewritePatternSet &patterns, const SparseTensorConversionOptions &options=SparseTensorConversionOptions())
Sets up sparse tensor conversion rules.
std::unique_ptr< Pass > createSparseTensorConversionPass()
SparseVectorizationStrategy vectorizationStrategy
static llvm::ManagedStatic< PassManagerOptions > options
SparseVectorizationStrategy
Defines a vectorization strategy.
SparseTensorConversionOptions(SparseToSparseConversionStrategy s2s)
SparsificationOptions(SparseParallelizationStrategy p, SparseVectorizationStrategy v, unsigned vl, bool e, bool vla)
void populateSparseTensorRewriting(RewritePatternSet &patterns)
void populateSparsificationPatterns(RewritePatternSet &patterns, const SparsificationOptions &options=SparsificationOptions())
Sets up sparsification rewriting rules with the given options.
SparseParallelizationStrategy parallelizationStrategy
Options for the Sparsification pass.
SparseParallelizationStrategy sparseParallelizationStrategy(int32_t flag)
Converts command-line parallelization flag to the strategy enum.
SparseVectorizationStrategy sparseVectorizationStrategy(int32_t flag)
Converts command-line vectorization flag to the strategy enum.
std::unique_ptr< Pass > createSparsificationPass()
Options for analysis-enabled bufferization.