Skip to content

Commit

Permalink
Internal change
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 586312803
  • Loading branch information
apaszke authored and jax authors committed Nov 29, 2023
1 parent 458a896 commit ef65ba8
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions tests/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ jax_test(
shard_count = {
"cpu": 10,
"gpu": 10,
"tpu": 20,
"tpu": 40,
},
)

Expand Down Expand Up @@ -403,7 +403,7 @@ jax_test(
shard_count = {
"cpu": 30,
"gpu": 30,
"tpu": 20,
"tpu": 40,
},
)

Expand Down Expand Up @@ -1223,7 +1223,12 @@ jax_test(
"gpu": 10,
"tpu": 50,
},
tags = ["multiaccelerator"],
tags = [
"multiaccelerator",
"noasan",
"nomsan",
"notsan",
], # Times out under *SAN.
deps = [
"//jax:tree_util",
],
Expand Down

0 comments on commit ef65ba8

Please sign in to comment.