experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change build system to use disabled_backends to disable selected backends. NFC.

Browse Source 
PiperOrigin-RevId: 318378024
Change-Id: I67c6d5f90b111555e5944d76cbd47c22875c5078
This commit is contained in:
A. Unique TensorFlower 2020-06-25 17:07:11 -07:00 committed by TensorFlower Gardener
parent 68f26867e7
commit 46c3b18c79
3 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/client/lib/BUILD
View File

@ -441,7 +441,7 @@ cc_library(
xla_test( xla_test(
name = "self_adjoint_eig_test", name = "self_adjoint_eig_test",
srcs = ["self_adjoint_eig_test.cc"], srcs = ["self_adjoint_eig_test.cc"],
blacklisted_backends = [ disabled_backends = [
"cpu", "cpu",
"gpu", "gpu",
], ],

16
tensorflow/compiler/xla/tests/BUILD
View File

@ -381,7 +381,7 @@ xla_test(
timeout = "long", timeout = "long",
srcs = ["conv_depthwise_backprop_filter_test.cc"], srcs = ["conv_depthwise_backprop_filter_test.cc"],
# these backends do not natively handle batch group counts. # these backends do not natively handle batch group counts.
blacklisted_backends = [ disabled_backends = [
"gpu", "gpu",
"cpu", "cpu",
], ],
@ -405,7 +405,7 @@ xla_test(
name = "grouped_convolution_test", name = "grouped_convolution_test",
timeout = "long", timeout = "long",
srcs = ["grouped_convolution_test.cc"], srcs = ["grouped_convolution_test.cc"],
blacklisted_backends = [ disabled_backends = [
# disabled because it times out. # disabled because it times out.
"cpu", "cpu",
], ],
@ -490,7 +490,7 @@ xla_test(
xla_test( xla_test(
name = "xla_hlo_profile_test", name = "xla_hlo_profile_test",
srcs = ["xla_hlo_profile_test.cc"], srcs = ["xla_hlo_profile_test.cc"],
blacklisted_backends = [ disabled_backends = [
# Hlo profiles are not supported on the interpreter backend. # Hlo profiles are not supported on the interpreter backend.
"interpreter", "interpreter",
], ],
@ -1220,7 +1220,7 @@ xla_test(
xla_test( xla_test(
name = "batch_normalization_test", name = "batch_normalization_test",
srcs = ["batch_normalization_test.cc"], srcs = ["batch_normalization_test.cc"],
blacklisted_backends = [ disabled_backends = [
# BatchNorm HLOs are not handled by the interpreter backend, and the # BatchNorm HLOs are not handled by the interpreter backend, and the
# BatchNorm expander is not run on the interpreter. # BatchNorm expander is not run on the interpreter.
"interpreter", "interpreter",
@ -1886,7 +1886,7 @@ xla_test(
xla_test( xla_test(
name = "all_reduce_test", name = "all_reduce_test",
srcs = ["all_reduce_test.cc"], srcs = ["all_reduce_test.cc"],
blacklisted_backends = [ disabled_backends = [
# All reduce is not supported on the interpreter backend. # All reduce is not supported on the interpreter backend.
"interpreter", "interpreter",
], ],
@ -2062,7 +2062,7 @@ xla_test(
xla_test( xla_test(
name = "execution_profile_test", name = "execution_profile_test",
srcs = ["execution_profile_test.cc"], srcs = ["execution_profile_test.cc"],
blacklisted_backends = [ disabled_backends = [
# Execution profiles are not supported on the interpreter backend. # Execution profiles are not supported on the interpreter backend.
"interpreter", "interpreter",
], ],
@ -2081,7 +2081,7 @@ xla_test(
name = "execution_profile_test_with_xla_hlo_profile", name = "execution_profile_test_with_xla_hlo_profile",
srcs = ["execution_profile_test.cc"], srcs = ["execution_profile_test.cc"],
args = ["--xla_hlo_profile"], args = ["--xla_hlo_profile"],
blacklisted_backends = [ disabled_backends = [
# Hlo profiles are not supported on the interpreter backend. # Hlo profiles are not supported on the interpreter backend.
"interpreter", "interpreter",
], ],
@ -2306,7 +2306,7 @@ xla_test(
xla_test( xla_test(
name = "outfeed_in_nested_computation_test", name = "outfeed_in_nested_computation_test",
srcs = ["outfeed_in_nested_computation_test.cc"], srcs = ["outfeed_in_nested_computation_test.cc"],
blacklisted_backends = [ disabled_backends = [
# Outfeed ops are not supported on the interpreter backend. # Outfeed ops are not supported on the interpreter backend.
"interpreter", "interpreter",
], ],

6
tensorflow/compiler/xla/tests/build_defs.bzl
View File

@ -34,7 +34,7 @@ def xla_test(
deps, deps,
xla_test_library_deps = [], xla_test_library_deps = [],
backends = [], backends = [],
blacklisted_backends = [], disabled_backends = [],
real_hardware_only = False, real_hardware_only = False,
args = [], args = [],
tags = [], tags = [],
@ -99,7 +99,7 @@ def xla_test(
backends: A list of backends to generate tests for. Supported values: "cpu", 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 "gpu". If this list is empty, the test will be generated for all supported
backends. 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. args: Test arguments for the target.
tags: Tags for the target. tags: Tags for the target.
copts: Additional copts to pass to the build. copts: Additional copts to pass to the build.
@ -121,7 +121,7 @@ def xla_test(
backends = [ backends = [
backend backend
for backend in backends for backend in backends
if backend not in blacklisted_backends if backend not in disabled_backends
] ]
native.cc_library( native.cc_library(