Changeset View
Changeset View
Standalone View
Standalone View
mlir/include/mlir/Dialect/Bufferization/Transforms/Passes.h
Show First 20 Lines • Show All 91 Lines • ▼ Show 20 Lines | |||||
/// Create a pass that tries to eliminate tensor.empty ops that are anchored on | /// Create a pass that tries to eliminate tensor.empty ops that are anchored on | ||||
/// insert_slice ops. | /// insert_slice ops. | ||||
std::unique_ptr<Pass> createEmptyTensorEliminationPass(); | std::unique_ptr<Pass> createEmptyTensorEliminationPass(); | ||||
/// Create a pass that bufferizes ops from the bufferization dialect. | /// Create a pass that bufferizes ops from the bufferization dialect. | ||||
std::unique_ptr<Pass> createBufferizationBufferizePass(); | std::unique_ptr<Pass> createBufferizationBufferizePass(); | ||||
/// Create a pass that resolves out-of-place tensor OpOperands with copies. | |||||
std::unique_ptr<Pass> createTensorCopyInsertionPass(); | |||||
std::unique_ptr<Pass> | |||||
createTensorCopyInsertionPass(const OneShotBufferizationOptions &options); | |||||
//===----------------------------------------------------------------------===// | //===----------------------------------------------------------------------===// | ||||
// Registration | // Registration | ||||
//===----------------------------------------------------------------------===// | //===----------------------------------------------------------------------===// | ||||
/// Register external models for AllocationOpInterface. | /// Register external models for AllocationOpInterface. | ||||
void registerAllocationOpInterfaceExternalModels(DialectRegistry ®istry); | void registerAllocationOpInterfaceExternalModels(DialectRegistry ®istry); | ||||
/// Generate the code for registering passes. | /// Generate the code for registering passes. | ||||
#define GEN_PASS_REGISTRATION | #define GEN_PASS_REGISTRATION | ||||
#include "mlir/Dialect/Bufferization/Transforms/Passes.h.inc" | #include "mlir/Dialect/Bufferization/Transforms/Passes.h.inc" | ||||
} // namespace bufferization | } // namespace bufferization | ||||
} // namespace mlir | } // namespace mlir | ||||
#endif // MLIR_DIALECT_BUFFERIZATION_TRANSFORMS_PASSES_H | #endif // MLIR_DIALECT_BUFFERIZATION_TRANSFORMS_PASSES_H |