MLIR  16.0.0git
Classes | Namespaces | Macros | Enumerations | Functions
Passes.h File Reference
#include "mlir/IR/PatternMatch.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Dialect/SparseTensor/Transforms/Passes.h.inc"
+ Include dependency graph for Passes.h:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mlir::SparsificationOptions
 Options for the Sparsification pass. More...
 
struct  mlir::SparseTensorConversionOptions
 SparseTensorConversion options. More...
 

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::bufferization
 

Macros

#define GEN_PASS_REGISTRATION
 Generate the code for registering passes. More...
 

Enumerations

enum  mlir::SparseParallelizationStrategy {
  mlir::SparseParallelizationStrategy::kNone, mlir::SparseParallelizationStrategy::kDenseOuterLoop, mlir::SparseParallelizationStrategy::kAnyStorageOuterLoop, mlir::SparseParallelizationStrategy::kDenseAnyLoop,
  mlir::SparseParallelizationStrategy::kAnyStorageAnyLoop
}
 Defines a parallelization strategy. More...
 
enum  mlir::SparseVectorizationStrategy { mlir::SparseVectorizationStrategy::kNone, mlir::SparseVectorizationStrategy::kDenseInnerLoop, mlir::SparseVectorizationStrategy::kAnyStorageInnerLoop }
 Defines a vectorization strategy. More...
 
enum  mlir::SparseToSparseConversionStrategy { mlir::SparseToSparseConversionStrategy::kAuto, mlir::SparseToSparseConversionStrategy::kViaCOO, mlir::SparseToSparseConversionStrategy::kDirect }
 Defines a strategy for implementing sparse-to-sparse conversion. More...
 

Functions

SparseParallelizationStrategy mlir::sparseParallelizationStrategy (int32_t flag)
 Converts command-line parallelization flag to the strategy enum. More...
 
SparseVectorizationStrategy mlir::sparseVectorizationStrategy (int32_t flag)
 Converts command-line vectorization flag to the strategy enum. More...
 
void mlir::populateSparsificationPatterns (RewritePatternSet &patterns, const SparsificationOptions &options=SparsificationOptions())
 Sets up sparsification rewriting rules with the given options. More...
 
std::unique_ptr< Passmlir::createSparsificationPass ()
 
std::unique_ptr< Passmlir::createSparsificationPass (const SparsificationOptions &options)
 
SparseToSparseConversionStrategy mlir::sparseToSparseConversionStrategy (int32_t flag)
 Converts command-line sparse2sparse flag to the strategy enum. More...
 
void mlir::populateSparseTensorConversionPatterns (TypeConverter &typeConverter, RewritePatternSet &patterns, const SparseTensorConversionOptions &options=SparseTensorConversionOptions())
 Sets up sparse tensor conversion rules. More...
 
std::unique_ptr< Passmlir::createSparseTensorConversionPass ()
 
std::unique_ptr< Passmlir::createSparseTensorConversionPass (const SparseTensorConversionOptions &options)
 
void mlir::populateSparseTensorRewriting (RewritePatternSet &patterns)
 
std::unique_ptr< Passmlir::createDenseBufferizationPass (const bufferization::OneShotBufferizationOptions &options)
 

Macro Definition Documentation

◆ GEN_PASS_REGISTRATION

#define GEN_PASS_REGISTRATION

Generate the code for registering passes.

Definition at line 155 of file Passes.h.