Skip to content

Commit c0c501d

Browse files
committed
uniformize names
1 parent e2e068b commit c0c501d

3 files changed

Lines changed: 5 additions & 13 deletions

File tree

ext/MatrixAlgebraKitCUDAExt/MatrixAlgebraKitCUDAExt.jl

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ using MatrixAlgebraKit: one!, zero!, uppertriangular!, lowertriangular!
66
using MatrixAlgebraKit: diagview, sign_safe
77
using MatrixAlgebraKit: CUSOLVER, LQViaTransposedQR, TruncationByValue, AbstractAlgorithm
88
using MatrixAlgebraKit: default_qr_algorithm, default_lq_algorithm, default_svd_algorithm, default_eig_algorithm, default_eigh_algorithm
9-
import MatrixAlgebraKit: geqrf!, ungqr!, unmqr!, _gpu_gesvd!, _gpu_Xgesvdp!, _gpu_Xgesvdr!, _gpu_gesvdj!, _gpu_geev!
9+
import MatrixAlgebraKit: geqrf!, ungqr!, unmqr!, gesvd!, gesvdp!, gesvdr!, gesvdj!, _gpu_geev!
1010
import MatrixAlgebraKit: _gpu_heevj!, _gpu_heevd!, _sylvester, svd_rank
1111
using CUDA, CUDA.CUBLAS
1212
using CUDA: i32
@@ -32,14 +32,6 @@ end
3232

3333
_gpu_geev!(A::StridedCuMatrix, D::StridedCuVector, V::StridedCuMatrix) =
3434
YACUSOLVER.Xgeev!(A, D, V)
35-
_gpu_gesvd!(A::StridedCuMatrix, S::StridedCuVector, U::StridedCuMatrix, Vᴴ::StridedCuMatrix) =
36-
YACUSOLVER.gesvd!(A, S, U, Vᴴ)
37-
_gpu_Xgesvdp!(A::StridedCuMatrix, S::StridedCuVector, U::StridedCuMatrix, Vᴴ::StridedCuMatrix; kwargs...) =
38-
YACUSOLVER.Xgesvdp!(A, S, U, Vᴴ; kwargs...)
39-
_gpu_Xgesvdr!(A::StridedCuMatrix, S::StridedCuVector, U::StridedCuMatrix, Vᴴ::StridedCuMatrix; kwargs...) =
40-
YACUSOLVER.Xgesvdr!(A, S, U, Vᴴ; kwargs...)
41-
_gpu_gesvdj!(A::StridedCuMatrix, S::StridedCuVector, U::StridedCuMatrix, Vᴴ::StridedCuMatrix; kwargs...) =
42-
YACUSOLVER.gesvdj!(A, S, U, Vᴴ; kwargs...)
4335

4436
_gpu_heevj!(A::StridedCuMatrix, Dd::StridedCuVector, V::StridedCuMatrix; kwargs...) =
4537
YACUSOLVER.heevj!(A, Dd, V; kwargs...)

ext/MatrixAlgebraKitCUDAExt/yacusolver.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ for (bname, fname, elty, relty) in
9898
end
9999
end
100100

101-
function Xgesvdp!(
101+
function gesvdp!(
102102
A::StridedCuMatrix{T},
103103
S::StridedCuVector = similar(A, real(T), min(size(A)...)),
104104
U::StridedCuMatrix{T} = similar(A, T, size(A, 1), min(size(A)...)),
@@ -165,7 +165,7 @@ function Xgesvdp!(
165165
end
166166
err = h_err_sigma[]
167167
if err > tol
168-
warn("Xgesvdp! did not attained requested tolerance: error = $err > tolerance = $tol")
168+
warn("gesvdp! did not attained requested tolerance: error = $err > tolerance = $tol")
169169
end
170170

171171
flag = @allowscalar dh.info[1]
@@ -269,7 +269,7 @@ for (bname, fname, elty, relty) in
269269
end
270270

271271
# Wrapper for randomized SVD
272-
function Xgesvdr!(
272+
function gesvdr!(
273273
A::StridedCuMatrix{T},
274274
S::StridedCuVector = similar(A, real(T), min(size(A)...)),
275275
U::StridedCuMatrix{T} = similar(A, T, size(A, 1), min(size(A)...)),

src/yalapack.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2303,7 +2303,7 @@ for (gesvd, gesdd, gesvdx, gejsv, gesvj, elty, relty) in
23032303
end
23042304
return (S, U, Vᴴ)
23052305
end
2306-
function gesvj!(
2306+
function gesvdj!(
23072307
A::AbstractMatrix{$elty},
23082308
S::AbstractVector{$relty} = similar(A, $relty, min(size(A)...)),
23092309
U::AbstractMatrix{$elty} = similar(

0 commit comments

Comments
 (0)