Skip async store reduction tests for int64/uint64 when x64 is disabled.

PiperOrigin-RevId: 845757351

Author: belitskiy

tests/mosaic/gpu_test.py

@@ -5402,6 +5402,10 @@ def body(
       reduction_op=("add", "min", "max", "inc", "dec", "and", "or", "xor"),
   )
   def test_async_store_reduction(self, dtype, reduction_op):
+
+    if not config.enable_x64.value and dtype in (jnp.int64, jnp.uint64):
+      self.skipTest("x64 support is disabled")
+
     # TODO(b/415721295):Clean up after the minimal jaxlib version is 0.8.2.
     if not hasattr(mgpu_dialect, "TMAReduction"):
       self.skipTest("The mgpu_dialect.TMAReduction attribute is required.")