test: Enable embedded dylib tests in Bazel Fix https://github.com/google/iree/issues/6748
diff --git a/iree/test/e2e/xla_ops/BUILD b/iree/test/e2e/xla_ops/BUILD index 9a4e62b..e3190d6 100644 --- a/iree/test/e2e/xla_ops/BUILD +++ b/iree/test/e2e/xla_ops/BUILD
@@ -95,75 +95,76 @@ target_backend = "cuda", ) -# TODO(#6748): Enable after issue is fixed. -# iree_check_single_backend_test_suite( -# name = "check_dylib_embedded-llvm-aot_dylib", -# srcs = enforce_glob( -# # keep sorted -# [ -# "abs.mlir", -# "add.mlir", -# "batch_norm_inference.mlir", -# "broadcast.mlir", -# "broadcast_add.mlir", -# "broadcast_in_dim.mlir", -# "clamp.mlir", -# "compare.mlir", -# "concatenate.mlir", -# "constant.mlir", -# "convert.mlir", -# "convolution.mlir", -# "cosine.mlir", -# "divide.mlir", -# "dot.mlir", -# "dot_general.mlir", -# "dynamic_slice.mlir", -# "dynamic_update_slice.mlir", -# "exponential.mlir", -# "exponential_minus_one.mlir", -# "fft.mlir", -# "finite.mlir", -# "floor.mlir", -# "gather.mlir", -# "iota.mlir", -# "log.mlir", -# "log_plus_one.mlir", -# "maximum.mlir", -# "minimum.mlir", -# "multiply.mlir", -# "negate.mlir", -# "pad.mlir", -# "reduce.mlir", -# "reduce_window.mlir", -# "remainder.mlir", -# "reshape.mlir", -# "reverse.mlir", -# "rsqrt.mlir", -# "scatter.mlir", -# "scatter_dynamic.mlir", -# "select.mlir", -# "sine.mlir", -# "slice.mlir", -# "sort.mlir", -# "sqrt.mlir", -# "subtract.mlir", -# "tanh.mlir", -# "torch_index_select.mlir", -# "transpose.mlir", -# "while.mlir", -# ], -# include = ["*.mlir"], -# exclude = [ -# "round.mlir", -# ], -# ), -# compiler_flags = [ -# "-iree-input-type=mhlo", -# "-iree-llvm-link-embedded=true", -# ], -# driver = "dylib", -# target_backend = "dylib-llvm-aot", -# ) +iree_check_single_backend_test_suite( + name = "check_dylib_embedded-llvm-aot_dylib", + srcs = enforce_glob( + # keep sorted + [ + "abs.mlir", + "add.mlir", + "batch_norm_inference.mlir", + "bitcast_convert.mlir", + "broadcast.mlir", + "broadcast_add.mlir", + "broadcast_in_dim.mlir", + "clamp.mlir", + "compare.mlir", + "concatenate.mlir", + "constant.mlir", + "convert.mlir", + "convolution.mlir", + "cosine.mlir", + "divide.mlir", + "dot.mlir", + "dot_general.mlir", + "dynamic_slice.mlir", + "dynamic_update_slice.mlir", + "exponential.mlir", + "exponential_minus_one.mlir", + "fft.mlir", + "finite.mlir", + "floor.mlir", + "gather.mlir", + "iota.mlir", + "log.mlir", + "log_plus_one.mlir", + "maximum.mlir", + "minimum.mlir", + "multiply.mlir", + "negate.mlir", + "pad.mlir", + "reduce.mlir", + "reduce_window.mlir", + "remainder.mlir", + "reshape.mlir", + "reverse.mlir", + "rng_uniform.mlir", + "rsqrt.mlir", + "scatter.mlir", + "scatter_dynamic.mlir", + "select.mlir", + "sine.mlir", + "slice.mlir", + "sort.mlir", + "sqrt.mlir", + "subtract.mlir", + "tanh.mlir", + "torch_index_select.mlir", + "transpose.mlir", + "while.mlir", + ], + include = ["*.mlir"], + exclude = [ + "round.mlir", + ], + ), + compiler_flags = [ + "-iree-input-type=mhlo", + "-iree-llvm-link-embedded=true", + ], + driver = "dylib", + target_backend = "dylib-llvm-aot", +) iree_check_single_backend_test_suite( name = "check_dylib-llvm-aot_dylib",
diff --git a/iree/test/e2e/xla_ops/CMakeLists.txt b/iree/test/e2e/xla_ops/CMakeLists.txt index 392dfa4..06ee2ad 100644 --- a/iree/test/e2e/xla_ops/CMakeLists.txt +++ b/iree/test/e2e/xla_ops/CMakeLists.txt
@@ -82,6 +82,71 @@ iree_check_single_backend_test_suite( NAME + check_dylib_embedded-llvm-aot_dylib + SRCS + "abs.mlir" + "add.mlir" + "batch_norm_inference.mlir" + "bitcast_convert.mlir" + "broadcast.mlir" + "broadcast_add.mlir" + "broadcast_in_dim.mlir" + "clamp.mlir" + "compare.mlir" + "concatenate.mlir" + "constant.mlir" + "convert.mlir" + "convolution.mlir" + "cosine.mlir" + "divide.mlir" + "dot.mlir" + "dot_general.mlir" + "dynamic_slice.mlir" + "dynamic_update_slice.mlir" + "exponential.mlir" + "exponential_minus_one.mlir" + "fft.mlir" + "finite.mlir" + "floor.mlir" + "gather.mlir" + "iota.mlir" + "log.mlir" + "log_plus_one.mlir" + "maximum.mlir" + "minimum.mlir" + "multiply.mlir" + "negate.mlir" + "pad.mlir" + "reduce.mlir" + "reduce_window.mlir" + "remainder.mlir" + "reshape.mlir" + "reverse.mlir" + "rng_uniform.mlir" + "rsqrt.mlir" + "scatter.mlir" + "scatter_dynamic.mlir" + "select.mlir" + "sine.mlir" + "slice.mlir" + "sort.mlir" + "sqrt.mlir" + "subtract.mlir" + "tanh.mlir" + "torch_index_select.mlir" + "transpose.mlir" + "while.mlir" + TARGET_BACKEND + "dylib-llvm-aot" + DRIVER + "dylib" + COMPILER_FLAGS + "-iree-input-type=mhlo" + "-iree-llvm-link-embedded=true" +) + +iree_check_single_backend_test_suite( + NAME check_dylib-llvm-aot_dylib SRCS "abs.mlir" @@ -267,70 +332,3 @@ ) ### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ### - -## TODO(#6748): This is moved below the `bazel-to-cmake-preserves` -## line till bazel builds can pass in this mode. Once the bazel builds -## are fixed these can be just deleted (and the auto-generated version -## can be used) -iree_check_single_backend_test_suite( - NAME - check_dylib_embedded-llvm-aot_dylib - SRCS - "abs.mlir" - "add.mlir" - "batch_norm_inference.mlir" - "broadcast.mlir" - "broadcast_add.mlir" - "broadcast_in_dim.mlir" - "clamp.mlir" - "compare.mlir" - "concatenate.mlir" - "constant.mlir" - "convert.mlir" - "convolution.mlir" - "cosine.mlir" - "divide.mlir" - "dot.mlir" - "dot_general.mlir" - "dynamic_slice.mlir" - "dynamic_update_slice.mlir" - "exponential.mlir" - "exponential_minus_one.mlir" - "fft.mlir" - "finite.mlir" - "floor.mlir" - "gather.mlir" - "iota.mlir" - "log.mlir" - "log_plus_one.mlir" - "maximum.mlir" - "minimum.mlir" - "multiply.mlir" - "negate.mlir" - "pad.mlir" - "reduce.mlir" - "reduce_window.mlir" - "remainder.mlir" - "reshape.mlir" - "reverse.mlir" - "rsqrt.mlir" - "scatter.mlir" - "scatter_dynamic.mlir" - "select.mlir" - "sine.mlir" - "slice.mlir" - "sort.mlir" - "sqrt.mlir" - "subtract.mlir" - "tanh.mlir" - "torch_index_select.mlir" - "transpose.mlir" - "while.mlir" - TARGET_BACKEND - "dylib-llvm-aot" - DRIVER - "dylib" - COMPILER_FLAGS - "-iree-input-type=mhlo" - "-iree-llvm-link-embedded=true" -)