fix test skipping

Signed-off-by: Jeremy Berchtold <[email protected]>

Author: jberchtold-nvidia

tests/jax/test_custom_call_compute.py

@@ -1925,7 +1925,7 @@ def test_grouped_dense_grad_fp8(self, fwd_bwd_dtype, scaling_mode, input_shape):
         assert_allclose(prim_wgrad, ref_wgrad, dtype=bwd_dtype)
         assert_allclose(prim_dbias, ref_dbias, dtype=dtype)
 
-
+@pytest.mark.skipif(not is_fp4_supported, reason=fp4_unsupported_reason)
 class TestFFICompatibility:
 
     HLO_DIR = os.path.join(os.path.dirname(__file__), "ffi_hlo")
@@ -2077,7 +2077,6 @@ def _make_args_based_on_input_tensor_shape_and_dtype(self, stablehlo_text: str):
             parsed_args.append(jnp.ones(shape, dtype=dtype))
         return parsed_args
 
-    @pytest.mark.skipif(is_fp4_supported, reason=fp4_unsupported_reason)
     def test_ffi_compatibility(self, ffi_hlo_name):
         """Tests that the current FFI bindings are compatible with the provided HLO and there are no API mismatches."""
         from jax.extend.backend import get_backend