Skip to content

Commit 147ce73

Browse files
metaflowGoogle-ML-Automation
authored andcommitted
[XLA:GPU] add lit deviceless tests
PiperOrigin-RevId: 900763193
1 parent e742f5b commit 147ce73

File tree

1 file changed

+14
-5
lines changed
  • xla/backends/gpu/codegen/triton/tests

1 file changed

+14
-5
lines changed

xla/backends/gpu/codegen/triton/tests/BUILD

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ load(
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")
1010
load("//xla:xla.default.bzl", "xla_cc_binary", "xla_cc_test")
1111
load("//xla/tests:build_defs.bzl", "xla_test")
1212
load("//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+
5365
lit_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",

0 commit comments

Comments
 (0)