32 #define GEN_PASS_DEF_TOSATOLINALG
33 #include "mlir/Conversion/Passes.h.inc"
39 struct TosaToLinalg :
public impl::TosaToLinalgBase<TosaToLinalg> {
43 .
insert<arith::ArithDialect, linalg::LinalgDialect, math::MathDialect,
44 index::IndexDialect, tensor::TensorDialect, scf::SCFDialect>();
47 void runOnOperation()
override {
66 FunctionOpInterface func = getOperation();
75 return std::make_unique<TosaToLinalg>();
80 const TosaToLinalgNamedOptions &tosaToLinalgNamedOptions,
81 tosa::TosaValidationOptions
const &validationOptions) {
83 if (!
options.disableTosaDecompositions)
94 {
options.aggressiveReduceConstant}));
96 pm.
addPass(tosa::createTosaValidation(validationOptions));
106 "tosa-to-linalg-pipeline",
107 "The default pipeline for converting TOSA operators to the equivalent "
108 "operations using the tensor operations in LinAlg as well as LinAlg "
111 TosaToLinalgOptions tosaToLinalgOptions;
112 TosaToLinalgNamedOptions tosaToLinalgNamedOptions;
114 tosaToLinalgNamedOptions,
116 {tosa::TosaProfileEnum::BaseInference,
118 tosa::TosaLevelEnum::EightK});
static MLIRContext * getContext(OpFoldResult val)
static llvm::ManagedStatic< PassManagerOptions > options
This class describes a specific conversion target.
void addLegalOp(OperationName op)
Register the given operations as legal.
void addLegalDialect(StringRef name, Names... names)
Register the operations of the given dialects as legal.
void markUnknownOpDynamicallyLegal(const DynamicLegalityCallbackFn &fn)
Register unknown operations as dynamically legal.
void addIllegalDialect(StringRef name, Names... names)
Register the operations of the given dialects as illegal, i.e.
The DialectRegistry maps a dialect namespace to a constructor for the matching dialect.
This class represents a pass manager that runs passes on either a specific operation type,...
void addPass(std::unique_ptr< Pass > pass)
Add the given pass to this pass manager.
void addNestedPass(std::unique_ptr< Pass > pass)
Add the given pass to a nested pass manager for the given operation kind OpT.
Operation is the basic unit of execution within MLIR.
void populateTosaToLinalgConversionPatterns(RewritePatternSet *patterns)
Populates conversion passes from TOSA dialect to Linalg dialect.
void addTosaToLinalgPasses(OpPassManager &pm, const TosaToLinalgOptions &options, const TosaToLinalgNamedOptions &tosaToLinalgNamedOptions=TosaToLinalgNamedOptions(), tosa::TosaValidationOptions const &validationOptions={ tosa::TosaProfileEnum::Undefined, false, tosa::TosaLevelEnum::None})
Populates passes to convert from TOSA to Linalg on buffers.
std::unique_ptr< Pass > createTosaMakeBroadcastablePass()
std::unique_ptr< Pass > createTosaToLinalgNamed(const TosaToLinalgNamedOptions &options=TosaToLinalgNamedOptions())
std::unique_ptr< Pass > createTosaToLinalg()
std::unique_ptr< Pass > createTosaOptionalDecompositions()
void registerTosaToLinalgPipelines()
Populates TOSA to linalg pipelines Currently, this includes only the "tosa-to-linalg-pipeline".
std::unique_ptr< Pass > createTosaLayerwiseConstantFoldPass()
std::unique_ptr< Pass > createTosaInferShapesPass()
Include the generated interface declarations.
LogicalResult applyFullConversion(ArrayRef< Operation * > ops, const ConversionTarget &target, const FrozenRewritePatternSet &patterns, ConversionConfig config=ConversionConfig())
Apply a complete conversion on the given operations, and all nested operations.
std::unique_ptr< Pass > createCanonicalizerPass()
Creates an instance of the Canonicalizer pass, configured with default settings (which can be overrid...
bool failed(LogicalResult result)
Utility function that returns true if the provided LogicalResult corresponds to a failure value.
PassPipelineRegistration provides a global initializer that registers a Pass pipeline builder routine...