Add TosaToMLProgram to the tosa input conversion pass pipeline (#16281)

Fixes [16251](#16251)
iree-org · Feb 1, 2024 · 8e1eca9 · 8e1eca9
1 parent dda1647
commit 8e1eca9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/compiler/plugins/input/TOSA/tosa-iree/InputConversion/BUILD.bazel b/compiler/plugins/input/TOSA/tosa-iree/InputConversion/BUILD.bazel
@@ -83,6 +83,7 @@ iree_compiler_cc_library(
         "@llvm-project//mlir:TosaDialect",
         "@llvm-project//mlir:TosaToArith",
         "@llvm-project//mlir:TosaToLinalg",
+        "@llvm-project//mlir:TosaToMLProgram",
         "@llvm-project//mlir:TosaToSCF",
         "@llvm-project//mlir:TosaToTensor",
         "@llvm-project//mlir:Transforms",

diff --git a/compiler/plugins/input/TOSA/tosa-iree/InputConversion/CMakeLists.txt b/compiler/plugins/input/TOSA/tosa-iree/InputConversion/CMakeLists.txt
@@ -71,6 +71,7 @@ iree_cc_library(
     MLIRTosaDialect
     MLIRTosaToArith
     MLIRTosaToLinalg
+    MLIRTosaToMLProgram
     MLIRTosaToSCF
     MLIRTosaToTensor
     MLIRTransforms

diff --git a/compiler/plugins/input/TOSA/tosa-iree/InputConversion/Passes.cpp b/compiler/plugins/input/TOSA/tosa-iree/InputConversion/Passes.cpp
@@ -9,6 +9,7 @@
 #include "iree/compiler/InputConversion/Common/Passes.h"
 #include "mlir/Conversion/TosaToArith/TosaToArith.h"
 #include "mlir/Conversion/TosaToLinalg/TosaToLinalg.h"
+#include "mlir/Conversion/TosaToMLProgram/TosaToMLProgram.h"
 #include "mlir/Conversion/TosaToSCF/TosaToSCF.h"
 #include "mlir/Conversion/TosaToTensor/TosaToTensor.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
@@ -31,6 +32,7 @@ void registerTOSAConversionPassPipeline() {
 
 // Prepare TOSA for use as an input to the Flow dialect.
 void buildTOSAInputConversionPassPipeline(OpPassManager &passManager) {
+  passManager.addPass(mlir::createTosaToMLProgram());
   // Currently we don't handle SCF ops well and have to convert them all to CFG.
   // In the future it would be nice if we could have all of flow be both scf
   // and cfg compatible.