Skip to content

Commit 509de78

Browse files
gulsumgudukbayRuturaj4
authored andcommitted
pallas/ops_test health check suppressed for test_select_n
1 parent 8a92b39 commit 509de78

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/pallas/ops_test.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,8 @@
5151

5252
import hypothesis.extra.numpy as hnp
5353
import hypothesis.strategies as hps
54+
from hypothesis import settings
55+
5456

5557
# There are many inherited redefinitions of _
5658
# ruff: noqa: F811
@@ -490,6 +492,7 @@ def kernel(x_ref, o_ref):
490492
# TODO(sharadmv): test rank < 2, size < 2
491493
@hp.given(select_n_strategy(max_cases=2, min_rank=2, max_rank=4,
492494
min_size_exp=1))
495+
@settings(suppress_health_check=[hp.HealthCheck.differing_executors])
493496
def test_select_n(self, args):
494497
if jtu.test_device_matches(["gpu"]):
495498
self.skipTest("TODO: error on GPU, lowering bug for select_n")

0 commit comments

Comments
 (0)