Reduce shard count for tests
Change: 132573817
This commit is contained in:
parent
3501c2fff5
commit
71e3186fd3
@ -262,7 +262,6 @@ cuda_py_tests(
|
|||||||
srcs = [
|
srcs = [
|
||||||
"batch_matrix_band_part_op_test.py",
|
"batch_matrix_band_part_op_test.py",
|
||||||
"conv_ops_3d_test.py",
|
"conv_ops_3d_test.py",
|
||||||
"cwise_ops_test.py",
|
|
||||||
"embedding_ops_test.py",
|
"embedding_ops_test.py",
|
||||||
"linalg_grad_test.py",
|
"linalg_grad_test.py",
|
||||||
"svd_op_test.py",
|
"svd_op_test.py",
|
||||||
@ -270,8 +269,18 @@ cuda_py_tests(
|
|||||||
additional_deps = [
|
additional_deps = [
|
||||||
"//tensorflow:tensorflow_py",
|
"//tensorflow:tensorflow_py",
|
||||||
],
|
],
|
||||||
|
shard_count = 20,
|
||||||
|
)
|
||||||
|
|
||||||
|
cuda_py_test(
|
||||||
|
name = "cwise_ops_test",
|
||||||
|
size = "medium",
|
||||||
|
srcs = ["cwise_ops_test.py"],
|
||||||
|
additional_deps = [
|
||||||
|
"//tensorflow:tensorflow_py",
|
||||||
|
],
|
||||||
shard_count = 50,
|
shard_count = 50,
|
||||||
tags = ["notap"], # b/30226163
|
tags = ["notap"], # http://b/30226163
|
||||||
)
|
)
|
||||||
|
|
||||||
filegroup(
|
filegroup(
|
||||||
|
Loading…
x
Reference in New Issue
Block a user