ROCm
diff --git a/‎jax/_src/export/_export.py‎
Lines changed: 1 addition & 0 deletions b/‎jax/_src/export/_export.py‎
Lines changed: 1 addition & 0 deletions
@@ -998,6 +998,7 @@ def _check_lowering(lowering) -> None:
     "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_sgehrd_ffi", "lapack_dgehrd_ffi", "lapack_cgehrd_ffi", "lapack_zgehrd_ffi",
+    "lapack_sgees_ffi", "lapack_dgees_ffi", "lapack_cgees_ffi", "lapack_zgees_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.
Original file line number	Diff line number	Diff line change
`@@ -998,6 +998,7 @@ def _check_lowering(lowering) -> None:`
`998`	`998`	`"lapack_sgesdd_ffi", "lapack_dgesdd_ffi", "lapack_cgesdd_ffi", "lapack_zgesdd_ffi",`
`999`	`999`	`"lapack_sgetrf_ffi", "lapack_dgetrf_ffi", "lapack_cgetrf_ffi", "lapack_zgetrf_ffi",`
`1000`	`1000`	`"lapack_sgehrd_ffi", "lapack_dgehrd_ffi", "lapack_cgehrd_ffi", "lapack_zgehrd_ffi",`
	`1001`	`+ "lapack_sgees_ffi", "lapack_dgees_ffi", "lapack_cgees_ffi", "lapack_zgees_ffi",`
`1001`	`1002`	`]`
`1002`	`1003`	`# These are the JAX custom call target names that are guaranteed to be stable.`
`1003`	`1004`	`# Their backwards compatibility is tested by back_compat_test.py.`