Update error message for eigh.

NikosKokkinis · Sep 25, 2019 · 298d838 · 298d838
1 parent f62b9d5
commit 298d838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/lax_linalg.py b/jax/lax_linalg.py
@@ -212,7 +212,7 @@ def eigh_impl(operand, lower):
 
 def eigh_translation_rule(c, operand, lower):
   raise NotImplementedError(
-    "Symmetric eigendecomposition is only implemented on the CPU backend")
+    "Symmetric eigendecomposition is only implemented on the CPU and GPU backends")
 
 def eigh_abstract_eval(operand, lower):
   if isinstance(operand, ShapedArray):