MLIR: include/mlir/Dialect/SparseTensor/Transforms/Passes.h File Reference (original) (raw)
Functions
void
mlir::populateSparseAssembler (RewritePatternSet &patterns, bool directOut)
std::unique_ptr< Pass >
mlir::createSparseAssembler ()
std::unique_ptr< Pass >
mlir::createSparseAssembler (bool directOut)
void
mlir::populateSparseReinterpretMap (RewritePatternSet &patterns, ReinterpretMapScope scope)
std::unique_ptr< Pass >
mlir::createSparseReinterpretMapPass ()
std::unique_ptr< Pass >
mlir::createSparseReinterpretMapPass (ReinterpretMapScope scope)
void
mlir::populatePreSparsificationRewriting (RewritePatternSet &patterns)
std::unique_ptr< Pass >
mlir::createPreSparsificationRewritePass ()
void
mlir::populateSparsificationPatterns (RewritePatternSet &patterns, const SparsificationOptions &options=SparsificationOptions())
Sets up sparsification rewriting rules with the given options. More...
std::unique_ptr< Pass >
mlir::createSparsificationPass ()
std::unique_ptr< Pass >
mlir::createSparsificationPass (const SparsificationOptions &options)
void
mlir::populateStageSparseOperationsPatterns (RewritePatternSet &patterns)
Sets up StageSparseOperation rewriting rules. More...
std::unique_ptr< Pass >
mlir::createStageSparseOperationsPass ()
void
mlir::populateLowerSparseOpsToForeachPatterns (RewritePatternSet &patterns, bool enableRT, bool enableConvert)
std::unique_ptr< Pass >
mlir::createLowerSparseOpsToForeachPass ()
std::unique_ptr< Pass >
mlir::createLowerSparseOpsToForeachPass (bool enableRT, bool enableConvert)
void
mlir::populateLowerForeachToSCFPatterns (RewritePatternSet &patterns)
std::unique_ptr< Pass >
mlir::createLowerForeachToSCFPass ()
void
mlir::populateLowerSparseIterationToSCFPatterns (const TypeConverter &converter, RewritePatternSet &patterns)
std::unique_ptr< Pass >
mlir::createLowerSparseIterationToSCFPass ()
void
mlir::populateSparseTensorConversionPatterns (const TypeConverter &typeConverter, RewritePatternSet &patterns)
Sets up sparse tensor conversion rules. More...
std::unique_ptr< Pass >
mlir::createSparseTensorConversionPass ()
void
mlir::populateSparseTensorCodegenPatterns (const TypeConverter &typeConverter, RewritePatternSet &patterns, bool createSparseDeallocs, bool enableBufferInitialization)
Sets up sparse tensor codegen rules. More...
std::unique_ptr< Pass >
mlir::createSparseTensorCodegenPass ()
std::unique_ptr< Pass >
mlir::createSparseTensorCodegenPass (bool createSparseDeallocs, bool enableBufferInitialization)
void
mlir::populateSparseBufferRewriting (RewritePatternSet &patterns, bool enableBufferInitialization)
std::unique_ptr< Pass >
mlir::createSparseBufferRewritePass ()
std::unique_ptr< Pass >
mlir::createSparseBufferRewritePass (bool enableBufferInitialization)
void
mlir::populateSparseVectorizationPatterns (RewritePatternSet &patterns, unsigned vectorLength, bool enableVLAVectorization, bool enableSIMDIndex32)
Populates the given patterns list with vectorization rules. More...
std::unique_ptr< Pass >
mlir::createSparseVectorizationPass ()
std::unique_ptr< Pass >
mlir::createSparseVectorizationPass (unsigned vectorLength, bool enableVLAVectorization, bool enableSIMDIndex32)
void
mlir::populateSparseGPUCodegenPatterns (RewritePatternSet &patterns, unsigned numThreads)
void
mlir::populateSparseGPULibgenPatterns (RewritePatternSet &patterns, bool enableRT)
std::unique_ptr< Pass >
mlir::createSparseGPUCodegenPass ()
std::unique_ptr< Pass >
mlir::createSparseGPUCodegenPass (unsigned numThreads, bool enableRT)
void
mlir::populateStorageSpecifierToLLVMPatterns (const TypeConverter &converter, RewritePatternSet &patterns)
std::unique_ptr< Pass >
mlir::createStorageSpecifierToLLVMPass ()
bufferization::OneShotBufferizationOptions
mlir::getBufferizationOptionsForSparsification (bool analysisOnly)
std::unique_ptr< Pass >
mlir::createSparsificationAndBufferizationPass ()
std::unique_ptr< Pass >
mlir::createSparsificationAndBufferizationPass (const bufferization::OneShotBufferizationOptions &bufferizationOptions, const SparsificationOptions &sparsificationOptions, bool createSparseDeallocs, bool enableRuntimeLibrary, bool enableBufferInitialization, unsigned vectorLength, bool enableVLAVectorization, bool enableSIMDIndex32, bool enableGPULibgen, SparseEmitStrategy emitStrategy, SparseParallelizationStrategy parallelizationStrategy)
std::unique_ptr< Pass >