diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/gpu/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/service/gpu/tests/BUILD | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/gpu/tests/BUILD b/tensorflow/compiler/xla/service/gpu/tests/BUILD index a725533567..1f0436278c 100644 --- a/tensorflow/compiler/xla/service/gpu/tests/BUILD +++ b/tensorflow/compiler/xla/service/gpu/tests/BUILD @@ -223,3 +223,15 @@ tf_cc_test( "@com_google_absl//absl/strings", ], ) + +tf_cc_test( + name = "gpu_atomic_test", + srcs = ["gpu_atomic_test.cc"], + tags = tf_cuda_tests_tags(), + deps = [ + ":gpu_codegen_test", + "//tensorflow/compiler/xla/tests:filecheck", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + ], +) |