Skip to content

Commit 26618cd

Browse files
committed
skip jit on .device
1 parent c04816a commit 26618cd

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/test_funcs.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,14 +28,16 @@
2828

2929
lazy_xp_function(atleast_nd, static_argnames=("ndim", "xp"))
3030
lazy_xp_function(cov, static_argnames="xp")
31-
lazy_xp_function(create_diagonal, static_argnames=("offset", "xp"))
31+
# FIXME .device attribute https://github.com/data-apis/array-api-compat/pull/238
32+
lazy_xp_function(create_diagonal, jax_jit=False, static_argnames=("offset", "xp"))
3233
lazy_xp_function(expand_dims, static_argnames=("axis", "xp"))
3334
lazy_xp_function(kron, static_argnames="xp")
3435
lazy_xp_function(nunique, static_argnames="xp")
3536
lazy_xp_function(pad, static_argnames=("pad_width", "mode", "constant_values", "xp"))
3637
# FIXME calls in1d which calls xp.unique_values without size
3738
lazy_xp_function(setdiff1d, jax_jit=False, static_argnames=("assume_unique", "xp"))
38-
lazy_xp_function(sinc, static_argnames="xp")
39+
# FIXME .device attribute https://github.com/data-apis/array-api-compat/pull/238
40+
lazy_xp_function(sinc, jax_jit=False, static_argnames="xp")
3941

4042

4143
@pytest.mark.skip_xp_backend(Backend.SPARSE, reason="no expand_dims")

0 commit comments

Comments
 (0)