1*6e6352f4SJerry-Ge //===- TosaToMLProgramPass.cpp - Lowering Tosa to MLProgram Dialect--------===// 2*6e6352f4SJerry-Ge // 3*6e6352f4SJerry-Ge // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*6e6352f4SJerry-Ge // See https://llvm.org/LICENSE.txt for license information. 5*6e6352f4SJerry-Ge // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*6e6352f4SJerry-Ge // 7*6e6352f4SJerry-Ge //===----------------------------------------------------------------------===// 8*6e6352f4SJerry-Ge // 9*6e6352f4SJerry-Ge // This transformation pass legalizes the TOSA dialect to the MLProgram dialect. 10*6e6352f4SJerry-Ge // 11*6e6352f4SJerry-Ge //===----------------------------------------------------------------------===// 12*6e6352f4SJerry-Ge 13*6e6352f4SJerry-Ge #include "mlir/Conversion/TosaToMLProgram/TosaToMLProgram.h" 14*6e6352f4SJerry-Ge #include "mlir/Dialect/MLProgram/IR/MLProgram.h" 15*6e6352f4SJerry-Ge #include "mlir/Dialect/Tosa/IR/TosaOps.h" 16*6e6352f4SJerry-Ge #include "mlir/Dialect/Tosa/Transforms/Passes.h" 17*6e6352f4SJerry-Ge #include "mlir/IR/PatternMatch.h" 18*6e6352f4SJerry-Ge #include "mlir/Pass/PassManager.h" 19*6e6352f4SJerry-Ge #include "mlir/Transforms/DialectConversion.h" 20*6e6352f4SJerry-Ge 21*6e6352f4SJerry-Ge namespace mlir { 22*6e6352f4SJerry-Ge #define GEN_PASS_DEF_TOSATOMLPROGRAM 23*6e6352f4SJerry-Ge #include "mlir/Conversion/Passes.h.inc" 24*6e6352f4SJerry-Ge } // namespace mlir 25*6e6352f4SJerry-Ge 26*6e6352f4SJerry-Ge using namespace mlir; 27*6e6352f4SJerry-Ge using namespace tosa; 28*6e6352f4SJerry-Ge 29*6e6352f4SJerry-Ge namespace { 30*6e6352f4SJerry-Ge struct TosaToMLProgram : public impl::TosaToMLProgramBase<TosaToMLProgram> { 31*6e6352f4SJerry-Ge public: runOnOperation__anon56ef5e0d0111::TosaToMLProgram32*6e6352f4SJerry-Ge void runOnOperation() override { 33*6e6352f4SJerry-Ge auto *context = &getContext(); 34*6e6352f4SJerry-Ge auto moduleOp = getOperation(); 35*6e6352f4SJerry-Ge 36*6e6352f4SJerry-Ge RewritePatternSet patterns(context); 37*6e6352f4SJerry-Ge ConversionTarget target(*context); 38*6e6352f4SJerry-Ge target.addIllegalOp<tosa::VariableOp, tosa::VariableReadOp, 39*6e6352f4SJerry-Ge tosa::VariableWriteOp>(); 40*6e6352f4SJerry-Ge target.markUnknownOpDynamicallyLegal([](Operation *) { return true; }); 41*6e6352f4SJerry-Ge 42*6e6352f4SJerry-Ge mlir::tosa::populateTosaToMLProgramConversionPatterns(&patterns); 43*6e6352f4SJerry-Ge 44*6e6352f4SJerry-Ge if (failed(applyPartialConversion(moduleOp, target, std::move(patterns)))) 45*6e6352f4SJerry-Ge signalPassFailure(); 46*6e6352f4SJerry-Ge } 47*6e6352f4SJerry-Ge }; 48*6e6352f4SJerry-Ge } // namespace 49