bring eig gaugefix in the same shape

Author: lkdvos

src/common/gauge.jl

@@ -8,7 +8,8 @@ This is achieved by ensuring that the entry with the largest magnitude in `V` or
 is real and positive.
 """ gaugefix!
 
-function gaugefix!(V::AbstractMatrix)
+
+function gaugefix!(::Union{typeof(eig_full!), typeof(eigh_full!), typeof(gen_eig_full!)}, V::AbstractMatrix)
     for j in axes(V, 2)
         v = view(V, :, j)
         s = conj(sign(_argmaxabs(v)))

src/implementations/eig.jl

@@ -93,7 +93,7 @@ function eig_full!(A::AbstractMatrix, DV, alg::LAPACK_EigAlgorithm)
         YALAPACK.geevx!(A, D.diag, V; lapack_kwargs...)
     end
 
-    dogaugefix && (V = gaugefix!(V))
+    dogaugefix && (V = gaugefix!(eig_full!, V))
 
     return D, V
 end
@@ -153,7 +153,7 @@ function eig_full!(A::AbstractMatrix, DV, alg::GPU_EigAlgorithm)
         _gpu_geev!(A, D.diag, V)
     end
 
-    dogaugefix && (V = gaugefix!(V))
+    dogaugefix && (V = gaugefix!(eig_full!, V))
 
     return D, V
 end

src/implementations/eigh.jl

@@ -105,7 +105,7 @@ function eigh_full!(A::AbstractMatrix, DV, alg::LAPACK_EighAlgorithm)
         YALAPACK.heevx!(A, Dd, V; lapack_kwargs...)
     end
 
-    dogaugefix && (V = gaugefix!(V))
+    dogaugefix && (V = gaugefix!(eigh_full!, V))
 
     return D, V
 end
@@ -183,7 +183,7 @@ function eigh_full!(A::AbstractMatrix, DV, alg::GPU_EighAlgorithm)
         throw(ArgumentError("Unsupported eigh algorithm"))
     end
 
-    dogaugefix && (V = gaugefix!(V))
+    dogaugefix && (V = gaugefix!(eigh_full!, V))
 
     return D, V
 end

src/implementations/gen_eig.jl

@@ -69,7 +69,7 @@ function gen_eig_full!(A::AbstractMatrix, B::AbstractMatrix, WV, alg::LAPACK_Eig
         throw(ArgumentError("LAPACK_Expert is not supported for ggev"))
     end
 
-    dogaugefix && (V = gaugefix!(V))
+    dogaugefix && (V = gaugefix!(gen_eig_full!, V))
 
     return W, V
 end