Add pass to convert nchw convolutions to nhwc (iree-org#15)

Include partial support for propagating and cancelling transposes generated when converting from nchw to nhwc
raikonenfnu · Oct 14, 2022 · de2066b · de2066b
1 parent c21cb9a
commit de2066b
Show file tree

Hide file tree

Showing 7 changed files with 714 additions and 0 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -50,3 +50,6 @@
 [submodule "third_party/level-zero"]
 	path = third_party/level-zero
 	url = https://github.com/oneapi-src/level-zero
+[submodule "level-zero"]
+	path = level-zero
+	url = https://github.com/oneapi-src/level-zero
diff --git a/compiler/src/iree/compiler/Dialect/Flow/Transforms/BUILD b/compiler/src/iree/compiler/Dialect/Flow/Transforms/BUILD
@@ -37,6 +37,7 @@ iree_compiler_cc_library(
         "CollapseReductionDims.cpp",
         "Convert1X1FilterConv2DToMatmul.cpp",
         "ConvertConv2DToImg2Col.cpp",
+        "ConvertConvNchwToNhwc.cpp",
         "ConvertLinalgMatmulToMmt4D.cpp",
         "ConvertRegionToWorkgroups.cpp",
         "ConvertToFlow.cpp",

diff --git a/compiler/src/iree/compiler/Dialect/Flow/Transforms/CMakeLists.txt b/compiler/src/iree/compiler/Dialect/Flow/Transforms/CMakeLists.txt
@@ -34,6 +34,7 @@ iree_cc_library(
     "CleanupTensorShapes.cpp"
     "CollapseReductionDims.cpp"
     "Convert1X1FilterConv2DToMatmul.cpp"
+    "ConvertConvNchwToNhwc.cpp"
     "ConvertConv2DToImg2Col.cpp"
     "ConvertLinalgMatmulToMmt4D.cpp"
     "ConvertRegionToWorkgroups.cpp"