[Cleanup] Refactor BUILD file. (#5972)
This makes us easier to track and add tests.
diff --git a/iree/test/e2e/xla_ops/BUILD b/iree/test/e2e/xla_ops/BUILD
index 5049de5..bbe09d9 100644
--- a/iree/test/e2e/xla_ops/BUILD
+++ b/iree/test/e2e/xla_ops/BUILD
@@ -26,105 +26,21 @@
licenses = ["notice"], # Apache 2.0
)
-CHECK_TESTS = [
- "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",
- "exponential.mlir",
- "exponential_minus_one.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",
- "select.mlir",
- "sine.mlir",
- "slice.mlir",
- "sqrt.mlir",
- "subtract.mlir",
- "tanh.mlir",
- "torch_index_select.mlir",
- "transpose.mlir",
- "while.mlir",
-]
-
iree_check_single_backend_test_suite(
name = "check_cuda",
- 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",
- "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", frem has a bug in LLVM PTX backend for negative values.
- "reshape.mlir",
- "reverse.mlir",
- #"round.mlir",
- "rsqrt.mlir",
- "select.mlir",
- "sine.mlir",
- "slice.mlir",
- #"sort.mlir",
- "sqrt.mlir",
- "subtract.mlir",
- "tanh.mlir",
- "torch_index_select.mlir",
- "transpose.mlir",
- "while.mlir",
- ],
+ srcs = glob(
+ ["*.mlir"],
+ exclude = [
+ "concatenate.mlir",
+ "dot.mlir",
+ "dot_general.mlir",
+ "reduce.mlir",
+ "reduce_window.mlir",
+ "remainder.mlir", # frem has a bug in LLVM PTX backend for negative values.
+ "round.mlir",
+ "sort.mlir",
+ ],
+ ),
driver = "cuda",
tags = [
# CUDA cuInit fails with sanitizer on.
@@ -139,21 +55,40 @@
iree_check_single_backend_test_suite(
name = "check_dylib-llvm-aot_dylib",
- srcs = CHECK_TESTS + ["fft.mlir"], # TODO(hanchung): clean this up, like VMLA test suite.
+ srcs = glob(
+ ["*.mlir"],
+ exclude = [
+ "round.mlir",
+ "sort.mlir",
+ ],
+ ),
driver = "dylib",
target_backend = "dylib-llvm-aot",
)
iree_check_single_backend_test_suite(
name = "check_vmvx_vmvx",
- srcs = CHECK_TESTS + ["fft.mlir"], # TODO(hanchung): clean this up, like VMLA test suite.
+ srcs = glob(
+ ["*.mlir"],
+ exclude = [
+ "round.mlir",
+ "sort.mlir",
+ ],
+ ),
driver = "vmvx",
target_backend = "vmvx",
)
iree_check_single_backend_test_suite(
name = "check_vulkan-spirv_vulkan",
- srcs = CHECK_TESTS,
+ srcs = glob(
+ ["*.mlir"],
+ exclude = [
+ "fft.mlir", # TODO(GH-5444): Enable the test.
+ "round.mlir",
+ "sort.mlir",
+ ],
+ ),
driver = "vulkan",
target_backend = "vulkan-spirv",
)
diff --git a/iree/test/e2e/xla_ops/CMakeLists.txt b/iree/test/e2e/xla_ops/CMakeLists.txt
index 64dcfeb..d58cb97 100644
--- a/iree/test/e2e/xla_ops/CMakeLists.txt
+++ b/iree/test/e2e/xla_ops/CMakeLists.txt
@@ -10,50 +10,28 @@
iree_add_all_subdirs()
+file(GLOB _GLOB_X_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS *.mlir)
+file(GLOB _GLOB_CONCATENATE_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS concatenate.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_CONCATENATE_MLIR})
+file(GLOB _GLOB_DOT_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS dot.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_DOT_MLIR})
+file(GLOB _GLOB_DOT_GENERAL_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS dot_general.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_DOT_GENERAL_MLIR})
+file(GLOB _GLOB_REDUCE_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS reduce.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_REDUCE_MLIR})
+file(GLOB _GLOB_REDUCE_WINDOW_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS reduce_window.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_REDUCE_WINDOW_MLIR})
+file(GLOB _GLOB_REMAINDER_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS remainder.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_REMAINDER_MLIR})
+file(GLOB _GLOB_ROUND_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS round.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_ROUND_MLIR})
+file(GLOB _GLOB_SORT_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS sort.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_SORT_MLIR})
iree_check_single_backend_test_suite(
NAME
check_cuda
SRCS
- "abs.mlir"
- "add.mlir"
- "batch_norm_inference.mlir"
- "broadcast.mlir"
- "broadcast_add.mlir"
- "broadcast_in_dim.mlir"
- "clamp.mlir"
- "compare.mlir"
- "constant.mlir"
- "convert.mlir"
- "convolution.mlir"
- "cosine.mlir"
- "divide.mlir"
- "dynamic_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"
- "reshape.mlir"
- "reverse.mlir"
- "rsqrt.mlir"
- "select.mlir"
- "sine.mlir"
- "slice.mlir"
- "sqrt.mlir"
- "subtract.mlir"
- "tanh.mlir"
- "torch_index_select.mlir"
- "transpose.mlir"
- "while.mlir"
+ "${_GLOB_X_MLIR}"
TARGET_BACKEND
"cuda"
DRIVER
@@ -66,164 +44,50 @@
"requires-gpu-nvidia"
)
+file(GLOB _GLOB_X_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS *.mlir)
+file(GLOB _GLOB_ROUND_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS round.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_ROUND_MLIR})
+file(GLOB _GLOB_SORT_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS sort.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_SORT_MLIR})
iree_check_single_backend_test_suite(
NAME
check_dylib-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"
- "exponential.mlir"
- "exponential_minus_one.mlir"
- "fft.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"
- "select.mlir"
- "sine.mlir"
- "slice.mlir"
- "sqrt.mlir"
- "subtract.mlir"
- "tanh.mlir"
- "torch_index_select.mlir"
- "transpose.mlir"
- "while.mlir"
+ "${_GLOB_X_MLIR}"
TARGET_BACKEND
"dylib-llvm-aot"
DRIVER
"dylib"
)
+file(GLOB _GLOB_X_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS *.mlir)
+file(GLOB _GLOB_ROUND_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS round.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_ROUND_MLIR})
+file(GLOB _GLOB_SORT_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS sort.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_SORT_MLIR})
iree_check_single_backend_test_suite(
NAME
check_vmvx_vmvx
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"
- "exponential.mlir"
- "exponential_minus_one.mlir"
- "fft.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"
- "select.mlir"
- "sine.mlir"
- "slice.mlir"
- "sqrt.mlir"
- "subtract.mlir"
- "tanh.mlir"
- "torch_index_select.mlir"
- "transpose.mlir"
- "while.mlir"
+ "${_GLOB_X_MLIR}"
TARGET_BACKEND
"vmvx"
DRIVER
"vmvx"
)
+file(GLOB _GLOB_X_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS *.mlir)
+file(GLOB _GLOB_FFT_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS fft.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_FFT_MLIR})
+file(GLOB _GLOB_ROUND_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS round.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_ROUND_MLIR})
+file(GLOB _GLOB_SORT_MLIR LIST_DIRECTORIES false RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} CONFIGURE_DEPENDS sort.mlir)
+list(REMOVE_ITEM _GLOB_X_MLIR ${_GLOB_SORT_MLIR})
iree_check_single_backend_test_suite(
NAME
check_vulkan-spirv_vulkan
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"
- "exponential.mlir"
- "exponential_minus_one.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"
- "select.mlir"
- "sine.mlir"
- "slice.mlir"
- "sqrt.mlir"
- "subtract.mlir"
- "tanh.mlir"
- "torch_index_select.mlir"
- "transpose.mlir"
- "while.mlir"
+ "${_GLOB_X_MLIR}"
TARGET_BACKEND
"vulkan-spirv"
DRIVER
