From 316b4f02d84bdb57a5207657651af55a4cde83fa Mon Sep 17 00:00:00 2001 From: Adrian Kuegel Date: Thu, 23 Jan 2020 03:58:34 -0800 Subject: [PATCH] Annotate a test with tf_cuda_tests_tags(). This is needed because it depends on gpu_device. PiperOrigin-RevId: 291138452 Change-Id: I18b34b3e77e5fc14508fd3af7f63e1ff8212b6ea --- tensorflow/compiler/jit/BUILD | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/compiler/jit/BUILD b/tensorflow/compiler/jit/BUILD index e479f8e9bf5..f380032ca33 100644 --- a/tensorflow/compiler/jit/BUILD +++ b/tensorflow/compiler/jit/BUILD @@ -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: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_root.bzl", "tf_cuda_tests_tags") package( default_visibility = [":internal"], @@ -755,7 +756,7 @@ tf_cc_test( ], # TODO(b/141643254) Re-enable msan after fixing use-of-uninitialized-value # error. - tags = ["nomsan"], + tags = ["nomsan"] + tf_cuda_tests_tags(), deps = [ ":common", ":compilation_passes",