Skip to content

Commit 7fd32ec

Browse files
bchetiouiGoogle-ML-Automation
authored andcommitted
[Pallas/Mosaic GPU] Explicitly disable ops_test on Mosaic GPU pre-Hopper.
PiperOrigin-RevId: 735744473
1 parent b6da46e commit 7fd32ec

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
@@ -283,6 +283,9 @@ def setUp(self):
283283
if (jtu.test_device_matches(["cuda"]) and
284284
not jtu.is_cuda_compute_capability_at_least("8.0")):
285285
self.skipTest("Only works on GPUs with capability >= sm80")
286+
if (jtu.test_device_matches(["cuda"]) and use_mosaic_gpu and
287+
not jtu.is_cuda_compute_capability_at_least("9.0")):
288+
self.skipTest("Mosaic GPU requires capability >= sm90")
286289

287290
super().setUp()
288291

0 commit comments

Comments
 (0)