Skip to content

Commit

Permalink
Revert to initial formatting of CPU FFI Kernels list
Browse files Browse the repository at this point in the history
This list has accidentally been auto-formatted which has caused unnecessary conflicts for future PRs.

PiperOrigin-RevId: 668368321
  • Loading branch information
pparuzel authored and jax authors committed Aug 28, 2024
1 parent eaefabe commit b0bd933
Showing 1 changed file with 4 additions and 16 deletions.
20 changes: 4 additions & 16 deletions jax/_src/export/_export.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit b0bd933

Please sign in to comment.