Add type alias for BlasMat

Author: lkdvos

src/interface/eig.jl

@@ -87,18 +87,18 @@ See also [`eig_full(!)`](@ref eig_full) and [`eig_trunc(!)`](@ref eig_trunc).
 
 # Algorithm selection
 # -------------------
-# Default to LAPACK for `StridedMatrix{<:BlasFloat}`
+# Default to LAPACK for `YALAPACK.BlasMat`
 function default_algorithm(::typeof(eig_full!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return LAPACK_Expert(; kwargs...)
 end
 function default_algorithm(::typeof(eig_vals!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return LAPACK_Expert(; kwargs...)
 end
 
 function select_algorithm(::typeof(eig_trunc!), ::Type{A}, alg; trunc=nothing,
-                          kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                          kwargs...) where {A<:YALAPACK.BlasMat}
     alg_eig = select_algorithm(eig_full!, A, alg; kwargs...)
     return TruncatedAlgorithm(alg_eig, select_truncation(trunc))
 end

src/interface/eigh.jl

@@ -86,18 +86,18 @@ See also [`eigh_full(!)`](@ref eigh_full) and [`eigh_trunc(!)`](@ref eigh_trunc)
 
 # Algorithm selection
 # -------------------
-# Default to LAPACK for `StridedMatrix{<:BlasFloat}`
+# Default to LAPACK for `YALAPACK.BlasMat`
 function default_algorithm(::typeof(eigh_full!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return LAPACK_MultipleRelativelyRobustRepresentations(; kwargs...)
 end
 function default_algorithm(::typeof(eigh_vals!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return LAPACK_MultipleRelativelyRobustRepresentations(; kwargs...)
 end
 
 function select_algorithm(::typeof(eigh_trunc!), ::Type{A}, alg; trunc=nothing,
-                          kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                          kwargs...) where {A<:YALAPACK.BlasMat}
     alg_eigh = select_algorithm(eigh_full!, A, alg; kwargs...)
     return TruncatedAlgorithm(alg_eigh, select_truncation(trunc))
 end

src/interface/lq.jl

@@ -71,7 +71,7 @@ See also [`qr_full(!)`](@ref lq_full) and [`qr_compact(!)`](@ref lq_compact).
 for f in (:lq_full!, :lq_compact!, :lq_null!)
     @eval begin
         function default_algorithm(::typeof($f), ::Type{A};
-                                   kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                                   kwargs...) where {A<:YALAPACK.BlasMat}
             return LAPACK_HouseholderLQ(; kwargs...)
         end
     end

src/interface/polar.jl

@@ -61,10 +61,10 @@ end
 # Algorithm selection
 # -------------------
 function default_algorithm(::typeof(left_polar!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return PolarViaSVD(default_algorithm(svd_compact!, A; kwargs...))
         end
 function default_algorithm(::typeof(right_polar!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return PolarViaSVD(default_algorithm(svd_compact!, A; kwargs...))
 end

src/interface/qr.jl

@@ -71,7 +71,7 @@ See also [`lq_full(!)`](@ref lq_full) and [`lq_compact(!)`](@ref lq_compact).
 for f in (:qr_full!, :qr_compact!, :qr_null!)
     @eval begin
         function default_algorithm(::typeof($f), ::Type{A};
-                                   kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                                   kwargs...) where {A<:YALAPACK.BlasMat}
             return LAPACK_HouseholderQR(; kwargs...)
         end
     end

src/interface/schur.jl

@@ -52,10 +52,10 @@ See also [`eig_full(!)`](@ref eig_full) and [`eig_trunc(!)`](@ref eig_trunc).
 # Algorithm selection
 # -------------------
 function default_algorithm(::typeof(schur_full!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return default_algorithm(eig_full!, A; kwargs...)
 end
 function default_algorithm(::typeof(schur_vals!), ::Type{A};
-                           kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                           kwargs...) where {A<:YALAPACK.BlasMat}
     return default_algorithm(eig_vals!, A; kwargs...)
 end

src/interface/svd.jl

@@ -91,15 +91,15 @@ See also [`svd_full(!)`](@ref svd_full), [`svd_compact(!)`](@ref svd_compact) an
 # Algorithm selection
 # -------------------
 for f in (:svd_full!, :svd_compact!, :svd_vals!)
-    # Default to LAPACK SDD for `StridedMatrix{<:BlasFloat}`
+    # Default to LAPACK SDD for `YALAPACK.BlasMat`
     @eval function default_algorithm(::typeof($f), ::Type{A};
-                                     kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                                     kwargs...) where {A<:YALAPACK.BlasMat}
         return LAPACK_DivideAndConquer(; kwargs...)
     end
 end
 
 function select_algorithm(::typeof(svd_trunc!), ::Type{A}, alg; trunc=nothing,
-                          kwargs...) where {A<:StridedMatrix{<:BlasFloat}}
+                          kwargs...) where {A<:YALAPACK.BlasMat}
     alg_svd = select_algorithm(svd_compact!, A, alg; kwargs...)
     return TruncatedAlgorithm(alg_svd, select_truncation(trunc))
 end

src/yalapack.jl

@@ -16,6 +16,9 @@ using LinearAlgebra: BlasFloat, BlasReal, BlasComplex, BlasInt, Char, LAPACK,
 using LinearAlgebra.BLAS: @blasfunc, libblastrampoline
 using LinearAlgebra.LAPACK: chkfinite, chktrans, chkside, chkuplofinite, chklapackerror
 
+# type alias for matrices that are definitely supported by YALAPACK
+const BlasMat{T<:BlasFloat} = StridedMatrix{T}
+
 # LU factorisation
 for (getrf, getrs, elty) in ((:dgetrf_, :dgetrs_, :Float64),
                              (:sgetrf_, :sgetrs_, :Float32),