mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #10260 from jakevdp:schur-gpu

Browse Source 
PiperOrigin-RevId: 441515694
This commit is contained in:
jax authors 2022-04-13 10:23:51 -07:00
commit a7ec509465
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
jax/_src/lax/linalg.py
View File

@ -1816,8 +1816,7 @@ def _schur_impl(operand, *, compute_schur_vectors, sort_eig_vals,
select_callable=select_callable)
def _schur_translation_rule(ctx, avals_in, avals_out, operand, *,
compute_schur_vectors, sort_eig_vals):
def _schur_translation_rule(ctx, *args, **kwargs):
raise NotImplementedError(
"Schur decomposition is only implemented on the CPU backend.")