Remove test with non-power-of-2 cluster size (unsupported by triton)

PiperOrigin-RevId: 706688862
jax-ml · Dec 16, 2024 · 91a29fb · 91a29fb
1 parent 0a8bdd5
commit 91a29fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/cluster_test.py b/tests/cluster_test.py
@@ -40,7 +40,7 @@ def dummy_kernel(x_ptr, o_ptr):
 
 class ClusterTest(parameterized.TestCase):
 
-  @parameterized.parameters(1, 2, 3, 4, 8)
+  @parameterized.parameters(1, 2, 4, 8)
   def test_cluster(self, num_ctas):
     if 'h100' not in jax.devices()[0].device_kind.lower():
       self.skipTest('Clusters available only on H100s.')