File tree Expand file tree Collapse file tree 1 file changed +14
-5
lines changed
xla/backends/gpu/codegen/triton/tests Expand file tree Collapse file tree 1 file changed +14
-5
lines changed Original file line number Diff line number Diff line change 66 "@local_config_sycl//sycl:build_defs.bzl" ,
77 "if_sycl_is_configured" ,
88)
9- load ("//xla:lit.bzl" , "lit_device_test" )
9+ load ("//xla:lit.bzl" , "lit_device_test" , "lit_test_suite" )
1010load ("//xla:xla.default.bzl" , "xla_cc_binary" , "xla_cc_test" )
1111load ("//xla/tests:build_defs.bzl" , "xla_test" )
1212load ("//xla/tsl:tsl.bzl" , "if_google" )
@@ -50,14 +50,23 @@ xla_cc_binary(
5050 ]),
5151)
5252
53+ deviceless_tests = ["collectives/all_reduce.hlo" ]
54+
55+ lit_test_suite (
56+ name = "deviceless_tests" ,
57+ srcs = deviceless_tests ,
58+ cfg = "//xla:lit.cfg.py" ,
59+ tools = [
60+ "//xla/backends/gpu/codegen/triton/tools:hlo_to_xtileir" ,
61+ "@llvm-project//llvm:FileCheck" ,
62+ ],
63+ )
64+
5365lit_device_test (
5466 name = "tests" ,
5567 srcs = glob (
5668 include = ["**/*.hlo" ],
57- exclude = [
58- # TODO(b/502824167): Re-enable once the test is fixed.
59- "collectives/*.hlo" ,
60- ],
69+ exclude = deviceless_tests ,
6170 ),
6271 backends = [
6372 "h100" ,
You can’t perform that action at this time.
0 commit comments