diff options
-rw-r--r-- | tensorflow/python/kernel_tests/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index 8db55a9359..d41a68f8e6 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -1215,6 +1215,7 @@ cuda_py_test( additional_deps = [ "//tensorflow:tensorflow_py", ], + shard_count = 10, ) cuda_py_test( |