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

Annotate a test with tf_cuda_tests_tags().

Browse Source 
This is needed because it depends on gpu_device.

PiperOrigin-RevId: 291138452
Change-Id: I18b34b3e77e5fc14508fd3af7f63e1ff8212b6ea
This commit is contained in:
Adrian Kuegel 2020-01-23 03:58:34 -08:00 committed by TensorFlower Gardener
parent 837637a306
commit 316b4f02d8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/jit/BUILD
View File

@ -2,6 +2,7 @@ load("//tensorflow:tensorflow.bzl", "cc_header_only_library", "if_mlir", "tf_cc_
load("//tensorflow/stream_executor:build_defs.bzl", "if_cuda_or_rocm") load("//tensorflow/stream_executor:build_defs.bzl", "if_cuda_or_rocm")
load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library", "tf_jit_compilation_passes_extra_deps") load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library", "tf_jit_compilation_passes_extra_deps")
load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library") load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library")
load("//tensorflow/core/platform:build_config_root.bzl", "tf_cuda_tests_tags")
package( package(
default_visibility = [":internal"], default_visibility = [":internal"],
@ -755,7 +756,7 @@ tf_cc_test(
], ],
# TODO(b/141643254) Re-enable msan after fixing use-of-uninitialized-value # TODO(b/141643254) Re-enable msan after fixing use-of-uninitialized-value
# error. # error.
tags = ["nomsan"], tags = ["nomsan"] + tf_cuda_tests_tags(),
deps = [ deps = [
":common", ":common",
":compilation_passes", ":compilation_passes",