Cleanup

Author: kshyatt

src/implementations/eigh.jl

@@ -18,7 +18,7 @@ function check_hermitian(A; atol::Real = default_hermitian_tol(A), rtol::Real =
 end
 
 function check_input(::typeof(eigh_full!), A::AbstractMatrix, DV, alg::AbstractAlgorithm)
-    #check_hermitian(A, alg)
+    check_hermitian(A, alg)
     D, V = DV
     m = size(A, 1)
     @assert D isa Diagonal && V isa AbstractMatrix

src/pullbacks/eig.jl

@@ -55,8 +55,7 @@ function eig_pullback!(
 
         check_eig_cotangents(D, VᴴΔV; degeneracy_atol, gauge_atol)
 
-        VᴴΔV ./= conj.(transpose(D) .- D)
-        diagview(VᴴΔV) .= zero(eltype(VᴴΔV))
+        VᴴΔV .*= conj.(inv_safe.(transpose(D) .- D, degeneracy_atol))
 
         if !iszerotangent(ΔDmat)
             ΔDvec = diagview(ΔDmat)