diff options
author | 2018-09-25 08:07:53 -0700 | |
---|---|---|
committer | 2018-09-25 08:12:24 -0700 | |
commit | 6419058dc6d065c34cccf55b11a7ba02c2ef7b06 (patch) | |
tree | e5c473eee00bc7a2cb7eec5ed2e4ea67ed3f290d /tensorflow/contrib/fused_conv | |
parent | 76b384d315b56d7bab1854579ee834763a460ca7 (diff) |
Fix test tag to require a sm70 GPU.
PiperOrigin-RevId: 214439785
Diffstat (limited to 'tensorflow/contrib/fused_conv')
-rw-r--r-- | tensorflow/contrib/fused_conv/BUILD | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/tensorflow/contrib/fused_conv/BUILD b/tensorflow/contrib/fused_conv/BUILD index 0f0813c07f..9725233e7f 100644 --- a/tensorflow/contrib/fused_conv/BUILD +++ b/tensorflow/contrib/fused_conv/BUILD @@ -111,7 +111,6 @@ tf_gen_op_wrapper_py( cuda_py_test( name = "fused_conv2d_bias_activation_op_test", - size = "large", srcs = ["python/ops/fused_conv2d_bias_activation_op_test.py"], additional_deps = [ ":fused_conv_py", @@ -130,14 +129,12 @@ cuda_py_test( "//tensorflow/python:variables", ], tags = [ - "manual", - "requires_cudnn6", + "requires-gpu-sm70", ], ) cuda_py_test( name = "fused_conv2d_bias_activation_benchmark", - size = "large", srcs = ["python/ops/fused_conv2d_bias_activation_benchmark.py"], additional_deps = [ ":fused_conv_py", @@ -155,7 +152,6 @@ cuda_py_test( ], main = "python/ops/fused_conv2d_bias_activation_benchmark.py", tags = [ - "manual", - "requires_cudnn6", + "requires-gpu-sm70", ], ) |