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