Skip to content

Commit dda9095

Browse files
belitskiyGoogle-ML-Automation
authored andcommitted
Suppress "healthcheck too slow" for OpsTest.test_select_n under ASAN.
PiperOrigin-RevId: 845251115
1 parent dd62dd7 commit dda9095

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/pallas/ops_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -550,6 +550,8 @@ def kernel(x_ref, o_ref):
550550
# TODO(sharadmv): test rank < 2, size < 2
551551
@hp.given(select_n_strategy(max_cases=2, min_rank=2, max_rank=4,
552552
min_size_exp=1))
553+
@hp.settings(suppress_health_check=([hp.HealthCheck.too_slow]
554+
if jtu.is_asan() else []))
553555
def test_select_n(self, args):
554556
if jtu.test_device_matches(["gpu"]):
555557
self.skipTest("TODO: error on GPU, lowering bug for select_n")

0 commit comments

Comments
 (0)