Enable mhlo.while e2e/xla_ops test for linalg-on-tensors path (#5054)

iree-org · Mar 11, 2021 · baf8a2d · baf8a2d
1 parent 0f455f2
commit baf8a2d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/iree/test/e2e/xla_ops/BUILD b/iree/test/e2e/xla_ops/BUILD
@@ -185,7 +185,7 @@ iree_check_single_backend_test_suite(
         # https://github.com/google/iree/issues/4079
         # "torch_index_select.mlir",
         "transpose.mlir",
-        # "while.mlir",
+        "while.mlir",
     ],
     compiler_flags = [
         "-iree-flow-dispatch-linalg-on-tensors",
@@ -246,7 +246,7 @@ iree_check_single_backend_test_suite(
         # https://github.com/google/iree/issues/4079
         # "torch_index_select.mlir",
         "transpose.mlir",
-        # "while.mlir",
+        "while.mlir",
     ],
     compiler_flags = [
         "-iree-flow-dispatch-linalg-on-tensors",

diff --git a/iree/test/e2e/xla_ops/CMakeLists.txt b/iree/test/e2e/xla_ops/CMakeLists.txt
@@ -166,6 +166,7 @@ iree_check_single_backend_test_suite(
     "subtract.mlir"
     "tanh.mlir"
     "transpose.mlir"
+    "while.mlir"
   TARGET_BACKEND
     "dylib-llvm-aot"
   DRIVER
@@ -213,6 +214,7 @@ iree_check_single_backend_test_suite(
     "subtract.mlir"
     "tanh.mlir"
     "transpose.mlir"
+    "while.mlir"
   TARGET_BACKEND
     "vulkan-spirv"
   DRIVER