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

Fix arguments to schur translation rule

Browse Source
This commit is contained in:
Jake VanderPlas 2022-04-13 09:50:33 -07:00
parent 86c8446c00
commit 7bfc86e17f
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.")