Change build system to use disabled_backends to disable selected backends. NFC.
PiperOrigin-RevId: 318378024 Change-Id: I67c6d5f90b111555e5944d76cbd47c22875c5078
This commit is contained in:
parent
68f26867e7
commit
46c3b18c79
@ -441,7 +441,7 @@ cc_library(
|
||||
xla_test(
|
||||
name = "self_adjoint_eig_test",
|
||||
srcs = ["self_adjoint_eig_test.cc"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
"cpu",
|
||||
"gpu",
|
||||
],
|
||||
|
@ -381,7 +381,7 @@ xla_test(
|
||||
timeout = "long",
|
||||
srcs = ["conv_depthwise_backprop_filter_test.cc"],
|
||||
# these backends do not natively handle batch group counts.
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
"gpu",
|
||||
"cpu",
|
||||
],
|
||||
@ -405,7 +405,7 @@ xla_test(
|
||||
name = "grouped_convolution_test",
|
||||
timeout = "long",
|
||||
srcs = ["grouped_convolution_test.cc"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
# disabled because it times out.
|
||||
"cpu",
|
||||
],
|
||||
@ -490,7 +490,7 @@ xla_test(
|
||||
xla_test(
|
||||
name = "xla_hlo_profile_test",
|
||||
srcs = ["xla_hlo_profile_test.cc"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
# Hlo profiles are not supported on the interpreter backend.
|
||||
"interpreter",
|
||||
],
|
||||
@ -1220,7 +1220,7 @@ xla_test(
|
||||
xla_test(
|
||||
name = "batch_normalization_test",
|
||||
srcs = ["batch_normalization_test.cc"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
# BatchNorm HLOs are not handled by the interpreter backend, and the
|
||||
# BatchNorm expander is not run on the interpreter.
|
||||
"interpreter",
|
||||
@ -1886,7 +1886,7 @@ xla_test(
|
||||
xla_test(
|
||||
name = "all_reduce_test",
|
||||
srcs = ["all_reduce_test.cc"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
# All reduce is not supported on the interpreter backend.
|
||||
"interpreter",
|
||||
],
|
||||
@ -2062,7 +2062,7 @@ xla_test(
|
||||
xla_test(
|
||||
name = "execution_profile_test",
|
||||
srcs = ["execution_profile_test.cc"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
# Execution profiles are not supported on the interpreter backend.
|
||||
"interpreter",
|
||||
],
|
||||
@ -2081,7 +2081,7 @@ xla_test(
|
||||
name = "execution_profile_test_with_xla_hlo_profile",
|
||||
srcs = ["execution_profile_test.cc"],
|
||||
args = ["--xla_hlo_profile"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
# Hlo profiles are not supported on the interpreter backend.
|
||||
"interpreter",
|
||||
],
|
||||
@ -2306,7 +2306,7 @@ xla_test(
|
||||
xla_test(
|
||||
name = "outfeed_in_nested_computation_test",
|
||||
srcs = ["outfeed_in_nested_computation_test.cc"],
|
||||
blacklisted_backends = [
|
||||
disabled_backends = [
|
||||
# Outfeed ops are not supported on the interpreter backend.
|
||||
"interpreter",
|
||||
],
|
||||
|
@ -34,7 +34,7 @@ def xla_test(
|
||||
deps,
|
||||
xla_test_library_deps = [],
|
||||
backends = [],
|
||||
blacklisted_backends = [],
|
||||
disabled_backends = [],
|
||||
real_hardware_only = False,
|
||||
args = [],
|
||||
tags = [],
|
||||
@ -99,7 +99,7 @@ def xla_test(
|
||||
backends: A list of backends to generate tests for. Supported values: "cpu",
|
||||
"gpu". If this list is empty, the test will be generated for all supported
|
||||
backends.
|
||||
blacklisted_backends: A list of backends to NOT generate tests for.
|
||||
disabled_backends: A list of backends to NOT generate tests for.
|
||||
args: Test arguments for the target.
|
||||
tags: Tags for the target.
|
||||
copts: Additional copts to pass to the build.
|
||||
@ -121,7 +121,7 @@ def xla_test(
|
||||
backends = [
|
||||
backend
|
||||
for backend in backends
|
||||
if backend not in blacklisted_backends
|
||||
if backend not in disabled_backends
|
||||
]
|
||||
|
||||
native.cc_library(
|
||||
|
Loading…
x
Reference in New Issue
Block a user