Skip to content

Commit 4551979

Browse files
Skip nvfp4 test on unsupported arch
Signed-off-by: Jeremy Berchtold <[email protected]>
1 parent 04072e3 commit 4551979

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/jax/test_custom_call_compute.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2055,6 +2055,7 @@ def _make_args_based_on_input_tensor_shape_and_dtype(self, stablehlo_text: str):
20552055
))
20562056
return parsed_args
20572057

2058+
@pytest.mark.skipif(is_fp4_supported, reason=fp4_unsupported_reason)
20582059
def test_ffi_compatibility(self, ffi_hlo_name):
20592060
""" Tests that the current FFI bindings are compatible with the provided HLO and there are no API mismatches. """
20602061
from jax.extend.backend import get_backend

0 commit comments

Comments
 (0)