diff --git a/jax/_src/export/_export.py b/jax/_src/export/_export.py index 88e4f21546fe..54a4f6d5498e 100644 --- a/jax/_src/export/_export.py +++ b/jax/_src/export/_export.py @@ -923,22 +923,10 @@ def _check_lowering(lowering) -> None: "\n".join(not_implemented_msgs)) _CPU_FFI_KERNELS = [ - "lapack_spotrf_ffi", - "lapack_dpotrf_ffi", - "lapack_cpotrf_ffi", - "lapack_zpotrf_ffi", - "lapack_sgeqrf_ffi", - "lapack_dgeqrf_ffi", - "lapack_cgeqrf_ffi", - "lapack_zgeqrf_ffi", - "lapack_sgesdd_ffi", - "lapack_dgesdd_ffi", - "lapack_cgesdd_ffi", - "lapack_zgesdd_ffi", - "lapack_sgetrf_ffi", - "lapack_dgetrf_ffi", - "lapack_cgetrf_ffi", - "lapack_zgetrf_ffi", + "lapack_spotrf_ffi", "lapack_dpotrf_ffi", "lapack_cpotrf_ffi", "lapack_zpotrf_ffi", + "lapack_sgeqrf_ffi", "lapack_dgeqrf_ffi", "lapack_cgeqrf_ffi", "lapack_zgeqrf_ffi", + "lapack_sgesdd_ffi", "lapack_dgesdd_ffi", "lapack_cgesdd_ffi", "lapack_zgesdd_ffi", + "lapack_sgetrf_ffi", "lapack_dgetrf_ffi", "lapack_cgetrf_ffi", "lapack_zgetrf_ffi", ] # These are the JAX custom call target names that are guaranteed to be stable. # Their backwards compatibility is tested by back_compat_test.py.