Wrap generalized Hermitian eigenvalue routine

JuliaParallel · Feb 26, 2018 · fc534b9 · fc534b9
1 parent c24b490
commit fc534b9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/core/types.jl b/src/core/types.jl
@@ -60,6 +60,7 @@ eltype{T}(A::ElementalMatrix{T}) = T
 @enum UpperOrLower LOWER UPPER
 @enum LeftOrRight LEFT RIGHT
 @enum UnitOrNonUnit NON_UNIT UNIT
+@enum Pencil AXBX=1 ABX=2 BAX=3
 
 # Get MPIWorldComm
 function CommWorldValue()

diff --git a/src/lapack_like/spectral.jl b/src/lapack_like/spectral.jl
@@ -242,6 +242,14 @@ for (elty, ext) in ((:Float32, :s),
                 return w
             end
 
+            function eigvalsHermitian(pencil::Pencil, uplo::UpperOrLower, A::$mat{$elty}, B::$mat{$elty}, sort::SortType = ASCENDING)
+                w = $mat(real($elty))
+                ElError(ccall(($(string("ElHermitianGenDefEig", mattype, "_", ext)), libEl), Cuint,
+                    (Pencil, UpperOrLower, Ptr{Void}, Ptr{Void}, Ptr{Void}, SortType),
+                    pencil, uplo, A.obj, B.obj, w.obj, sort))
+                return w
+            end
+
             function eigHermitian(uplo::UpperOrLower, A::$mat{$elty}, sort::SortType = ASCENDING)
                 w = $mat(real($elty))
                 Z = $mat($elty)