[NFC][LinalgExt] Move tiling tests and implementations from IR to Tra…

…nsforms (iree-org#17216) This PR moves all tiling interface implementations and tests from `LinalgExt/IR` to `LinalgExt/Transforms`. It makes more sense to have tiling implementations in transforms, and this more closely aligns with the Linalg dialect upstream.
kumardeepakamd · May 9, 2024 · fc3561c · fc3561c
1 parent 778bf31
commit fc3561c
Show file tree

Hide file tree

Showing 10 changed files with 1,704 additions and 1,624 deletions.
diff --git a/compiler/src/iree/compiler/Dialect/LinalgExt/IR/LinalgExtOps.cpp b/compiler/src/iree/compiler/Dialect/LinalgExt/IR/LinalgExtOps.cpp
diff --git a/compiler/src/iree/compiler/Dialect/LinalgExt/IR/test/BUILD.bazel b/compiler/src/iree/compiler/Dialect/LinalgExt/IR/test/BUILD.bazel
@@ -17,10 +17,8 @@ iree_lit_test_suite(
     srcs = enforce_glob(
         [
             "canonicalize.mlir",
-            "distribution.mlir",
             "invalid.mlir",
             "roundtrip.mlir",
-            "tiling.mlir",
         ],
         include = ["*.mlir"],
     ),

diff --git a/compiler/src/iree/compiler/Dialect/LinalgExt/IR/test/CMakeLists.txt b/compiler/src/iree/compiler/Dialect/LinalgExt/IR/test/CMakeLists.txt
@@ -15,10 +15,8 @@ iree_lit_test_suite(
     lit
   SRCS
     "canonicalize.mlir"
-    "distribution.mlir"
     "invalid.mlir"
     "roundtrip.mlir"
-    "tiling.mlir"
   TOOLS
     FileCheck
     iree-opt

diff --git a/compiler/src/iree/compiler/Dialect/LinalgExt/Transforms/BUILD.bazel b/compiler/src/iree/compiler/Dialect/LinalgExt/Transforms/BUILD.bazel
@@ -38,6 +38,7 @@ iree_compiler_cc_library(
         "Passes.cpp",
         "SplitReduction.cpp",
         "TileAndDecomposeAttention.cpp",
+        "TilingInterfaceImpl.cpp",
     ],
     hdrs = [
         "Passes.h",

diff --git a/compiler/src/iree/compiler/Dialect/LinalgExt/Transforms/CMakeLists.txt b/compiler/src/iree/compiler/Dialect/LinalgExt/Transforms/CMakeLists.txt
@@ -34,6 +34,7 @@ iree_cc_library(
     "Passes.cpp"
     "SplitReduction.cpp"
     "TileAndDecomposeAttention.cpp"
+    "TilingInterfaceImpl.cpp"
   DEPS
     ::PassesIncGen
     IREEInputDialect