From baf8a2dae983683ca9b163d5a2c21578db9d6dca Mon Sep 17 00:00:00 2001 From: "Ahmed S. Taei" Date: Wed, 10 Mar 2021 20:54:58 -0800 Subject: [PATCH] Enable mhlo.while e2e/xla_ops test for linalg-on-tensors path (#5054) --- iree/test/e2e/xla_ops/BUILD | 4 ++-- iree/test/e2e/xla_ops/CMakeLists.txt | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/iree/test/e2e/xla_ops/BUILD b/iree/test/e2e/xla_ops/BUILD index 4967da1ef81b..bd6eff8b6d2a 100644 --- 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 index 90ed6bd88079..9a413975afb2 100644 --- 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