From b0bd9337c90e00747d264b8eb91b746d644e91af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20Paruzel?= Date: Wed, 28 Aug 2024 03:05:50 -0700 Subject: [PATCH] Revert to initial formatting of CPU FFI Kernels list This list has accidentally been auto-formatted which has caused unnecessary conflicts for future PRs. PiperOrigin-RevId: 668368321 --- jax/_src/export/_export.py | 20 ++++---------------- 1 file changed, 4 insertions(+), 16 deletions(-) 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.