Merge pull request #305 from Xilinx/jrickert.lower_to_linalg

Add Lowering for xten.resize, xten.grid_sample and xten.reduce_mean to linalg

Author: jorickert

include/xten/Conversion/Passes.td

@@ -33,8 +33,9 @@ def XTenNNToTosa : Pass<"xten-nn-to-tosa", "ModuleOp"> {
   let description = [{
   Partial conversion of XTenNN operations to TOSA.
 
-  This pass only converts XTenNN quantize and dequantize operations
-  who operate on signed integer tensors to TOSA.
+  This pass converts XTenNN quantize and dequantize operations
+  who operate on signed integer tensors to TOSA. It can also optionally
+  decompose XTenNN depth_to_space operations to TOSA.
 
   Currently, unsigned integer tensor support in TOSA does not allow
   lowerings of operations like:
@@ -49,6 +50,13 @@ def XTenNNToTosa : Pass<"xten-nn-to-tosa", "ModuleOp"> {
     "mlir::tosa::TosaDialect",
     "amd::xten_nn::XTenNNDialect"
   ];
+  let options = [
+    Option<"enableDepthToSpaceDecomposition",
+           "enable-depth-to-space-decomposition",
+           "bool",
+           /*default=*/"false",
+           "Enable decomposition of xten_nn.depth_to_space to TOSA.">,
+  ];
 }
 
 def ConvertXTenNNToLinalg : Pass<"convert-xtennn-to-linalg", "mlir::func::FuncOp"> {

include/xten/Conversion/XTenNNToTosaPass.h

@@ -19,6 +19,8 @@ namespace amd {
 namespace xten_nn {
 
 std::unique_ptr<mlir::Pass> createXTenNNToTOSAPass();
+std::unique_ptr<mlir::Pass>
+createXTenNNToTOSAPass(bool enableDepthToSpaceDecomposition);
 
 } // namespace xten_nn
 } // namespace amd