@@ -16,17 +16,19 @@ const ungqr! = orgqr!
1616
1717# Wrapper for SVD via QR Iteration
1818for (fname, elty, relty) in
19- ((:rocsolver_sgesvd, :Float32, :Float32),
20- (:rocsolver_dgesvd, :Float64, :Float64),
21- (:rocsolver_cgesvd, :ComplexF32, :Float32),
22- (:rocsolver_zgesvd, :ComplexF64, :Float64))
19+ (
20+ (:rocsolver_sgesvd, :Float32, :Float32),
21+ (:rocsolver_dgesvd, :Float64, :Float64),
22+ (:rocsolver_cgesvd, :ComplexF32, :Float32),
23+ (:rocsolver_zgesvd, :ComplexF64, :Float64),
24+ )
2325 @eval begin
24- # ! format: off
25- function gesvd!( A:: StridedROCMatrix{$elty} ,
26- S:: StridedROCVector{$relty} = similar(A, $ relty, min(size(A). .. )),
27- U:: StridedROCMatrix{$elty} = similar(A, $ elty, size(A, 1 ), min(size(A). .. )),
28- Vᴴ:: StridedROCMatrix{$elty} = similar(A, $ elty, min(size(A). .. ), size(A, 2 ) ))
29- # ! format: on
26+ function gesvd!(
27+ A:: StridedROCMatrix{$elty} ,
28+ S:: StridedROCVector{$relty} = similar(A, $ relty, min(size(A). .. )),
29+ U:: StridedROCMatrix{$elty} = similar(A, $ elty, size(A, 1 ), min(size(A). .. )),
30+ Vᴴ:: StridedROCMatrix{$elty} = similar(A, $ elty, min(size(A). .. ), size(A, 2 ))
31+ )
3032 chkstride1(A, U, Vᴴ, S)
3133 m, n = size(A)
3234 (m < n) && throw(ArgumentError(" rocSOLVER's gesvd requires m ≥ n"
@@ -72,13 +74,15 @@ for (fname, elty, relty) in
7274 ldu = max(1 , stride(U, 2 ))
7375 ldv = max(1 , stride(Vᴴ, 2 ))
7476
75- rwork = ROCArray{$ relty}(undef, minmn - 1 )
76- dh = rocBLAS. handle()
77+ rwork = ROCArray{$ relty}(undef, minmn - 1 )
78+ dh = rocBLAS. handle()
7779 dev_info = ROCVector{Cint}(undef, 1 )
78- rocSOLVER.$ fname(dh, jobu, jobvt, m, n,
79- A, lda, S, U, ldu, Vᴴ, ldv,
80- rwork, convert(rocSOLVER. rocblas_workmode, ' I'
81- dev_info)
80+ rocSOLVER.$ fname(
81+ dh, jobu, jobvt, m, n,
82+ A, lda, S, U, ldu, Vᴴ, ldv,
83+ rwork, convert(rocSOLVER. rocblas_workmode, ' I'
84+ dev_info
85+ )
8286 AMDGPU. unsafe_free!(rwork)
8387
8488 info = @allowscalar dev_info[1 ]
9195
9296# Wrapper for SVD via Jacobi
9397for (fname, elty, relty) in
94- ((:rocsolver_sgesvdj, :Float32, :Float32),
95- (:rocsolver_dgesvdj, :Float64, :Float64),
96- (:rocsolver_cgesvdj, :ComplexF32, :Float32),
97- (:rocsolver_zgesvdj, :ComplexF64, :Float64))
98+ (
99+ (:rocsolver_sgesvdj, :Float32, :Float32),
100+ (:rocsolver_dgesvdj, :Float64, :Float64),
101+ (:rocsolver_cgesvdj, :ComplexF32, :Float32),
102+ (:rocsolver_zgesvdj, :ComplexF64, :Float64),
103+ )
98104 @eval begin
99- # ! format: off
100- function gesvdj!(A:: StridedROCMatrix{$elty} ,
101- S:: StridedROCVector{$relty} = similar(A, $ relty, min(size(A). .. )),
102- U:: StridedROCMatrix{$elty} = similar(A, $ elty, size(A, 1 ), min(size(A). .. )),
103- Vᴴ:: StridedROCMatrix{$elty} = similar(A, $ elty, min(size(A). .. ), size(A, 2 ));
104- tol:: $relty = eps($ relty),
105- max_sweeps:: Int = 100 ,
106- )
107- # ! format: on
105+ function gesvdj!(
106+ A:: StridedROCMatrix{$elty} ,
107+ S:: StridedROCVector{$relty} = similar(A, $ relty, min(size(A). .. )),
108+ U:: StridedROCMatrix{$elty} = similar(A, $ elty, size(A, 1 ), min(size(A). .. )),
109+ Vᴴ:: StridedROCMatrix{$elty} = similar(A, $ elty, min(size(A). .. ), size(A, 2 ));
110+ tol:: $relty = eps($ relty),
111+ max_sweeps:: Int = 100 ,
112+ )
108113 chkstride1(A, U, Vᴴ, S)
109114 m, n = size(A)
110115 minmn = min(m, n)
@@ -149,21 +154,22 @@ for (fname, elty, relty) in
149154 lda = max(1 , stride(A, 2 ))
150155 ldu = max(1 , stride(U, 2 ))
151156 ldv = max(1 , stride(Vᴴ, 2 ))
152- dev_info = ROCVector{Cint}(undef, 1 )
157+ dev_info = ROCVector{Cint}(undef, 1 )
153158 dev_residual = ROCVector{$ relty}(undef, 1 )
154159 dev_n_sweeps = ROCVector{Cint}(undef, 1 )
155160
156161 dh = rocBLAS. handle()
157- rocSOLVER.$ fname(dh, jobu, jobvt, m, n, A, lda, tol,
158- dev_residual, max_sweeps, dev_n_sweeps,
159- S, U, ldu, Vᴴ, ldv, dev_info,
160- )
162+ rocSOLVER.$ fname(
163+ dh, jobu, jobvt, m, n, A, lda, tol,
164+ dev_residual, max_sweeps, dev_n_sweeps,
165+ S, U, ldu, Vᴴ, ldv, dev_info,
166+ )
161167
162168 info = @allowscalar dev_info[1 ]
163169 rocSOLVER. chkargsok(BlasInt(info))
164170
165- AMDGPU. unsafe_free!(dev_residual)
166- AMDGPU. unsafe_free!(dev_n_sweeps)
171+ AMDGPU. unsafe_free!(dev_residual)
172+ AMDGPU. unsafe_free!(dev_n_sweeps)
167173 return (S, U, Vᴴ)
168174 end
169175 end
@@ -476,15 +482,19 @@ end
476482# end
477483
478484for (heevd, heev, heevx, heevj, elty, relty) in
479- ((:(rocSOLVER. rocsolver_ssyevd), :(rocSOLVER. rocsolver_ssyev), :(rocSOLVER. rocsolver_ssyevx), :(rocSOLVER. rocsolver_ssyevj), :Float32, :Float32),
480- (:(rocSOLVER. rocsolver_dsyevd), :(rocSOLVER. rocsolver_dsyev), :(rocSOLVER. rocsolver_dsyevx), :(rocSOLVER. rocsolver_dsyevj), :Float64, :Float64),
481- (:(rocSOLVER. rocsolver_cheevd), :(rocSOLVER. rocsolver_cheev), :(rocSOLVER. rocsolver_cheevx), :(rocSOLVER. rocsolver_cheevj), :ComplexF32, :Float32),
482- (:(rocSOLVER. rocsolver_zheevd), :(rocSOLVER. rocsolver_zheev), :(rocSOLVER. rocsolver_zheevx), :(rocSOLVER. rocsolver_zheevj), :ComplexF64, :Float64))
485+ (
486+ (:(rocSOLVER. rocsolver_ssyevd), :(rocSOLVER. rocsolver_ssyev), :(rocSOLVER. rocsolver_ssyevx), :(rocSOLVER. rocsolver_ssyevj), :Float32, :Float32),
487+ (:(rocSOLVER. rocsolver_dsyevd), :(rocSOLVER. rocsolver_dsyev), :(rocSOLVER. rocsolver_dsyevx), :(rocSOLVER. rocsolver_dsyevj), :Float64, :Float64),
488+ (:(rocSOLVER. rocsolver_cheevd), :(rocSOLVER. rocsolver_cheev), :(rocSOLVER. rocsolver_cheevx), :(rocSOLVER. rocsolver_cheevj), :ComplexF32, :Float32),
489+ (:(rocSOLVER. rocsolver_zheevd), :(rocSOLVER. rocsolver_zheev), :(rocSOLVER. rocsolver_zheevx), :(rocSOLVER. rocsolver_zheevj), :ComplexF64, :Float64),
490+ )
483491 @eval begin
484- function heevd!(A:: StridedROCMatrix{$elty} ,
485- W:: StridedROCVector{$relty} ,
486- V:: StridedROCMatrix{$elty} ;
487- uplo:: Char = ' U'
492+ function heevd!(
493+ A:: StridedROCMatrix{$elty} ,
494+ W:: StridedROCVector{$relty} ,
495+ V:: StridedROCMatrix{$elty} ;
496+ uplo:: Char = ' U'
497+ )
488498 chkuplo(uplo)
489499 n = checksquare(A)
490500 lda = max(1 , stride(A, 2 ))
@@ -509,10 +519,12 @@ for (heevd, heev, heevx, heevj, elty, relty) in
509519 end
510520 return W, V
511521 end
512- function heev!(A:: StridedROCMatrix{$elty} ,
513- W:: StridedROCVector{$relty} ,
514- V:: StridedROCMatrix{$elty} ;
515- uplo:: Char = ' U'
522+ function heev!(
523+ A:: StridedROCMatrix{$elty} ,
524+ W:: StridedROCVector{$relty} ,
525+ V:: StridedROCMatrix{$elty} ;
526+ uplo:: Char = ' U'
527+ )
516528 chkuplo(uplo)
517529 n = checksquare(A)
518530 lda = max(1 , stride(A, 2 ))
@@ -537,11 +549,13 @@ for (heevd, heev, heevx, heevj, elty, relty) in
537549 end
538550 return W, V
539551 end
540- function heevx!(A:: StridedROCMatrix{$elty} ,
541- W:: StridedROCVector{$relty} ,
542- V:: StridedROCMatrix{$elty} ;
543- uplo:: Char = ' U'
544- kwargs... )
552+ function heevx!(
553+ A:: StridedROCMatrix{$elty} ,
554+ W:: StridedROCVector{$relty} ,
555+ V:: StridedROCMatrix{$elty} ;
556+ uplo:: Char = ' U'
557+ kwargs...
558+ )
545559 chkuplo(uplo)
546560 n = checksquare(A)
547561 lda = max(1 , stride(A, 2 ))
@@ -567,27 +581,29 @@ for (heevd, heev, heevx, heevj, elty, relty) in
567581 size(V) == (n, n) || throw(DimensionMismatch(" size mismatch between A and V"
568582 jobz = rocSOLVER. rocblas_evect_original
569583 end
570- dh = rocBLAS. handle()
571- abstol = - one($ relty)
572- nev = ROCVector{Cint}(undef, 1 )
573- ldv = max(1 , stride(V, 2 ))
574- ifail = ROCVector{Cint}(undef, n)
584+ dh = rocBLAS. handle()
585+ abstol = - one($ relty)
586+ nev = ROCVector{Cint}(undef, 1 )
587+ ldv = max(1 , stride(V, 2 ))
588+ ifail = ROCVector{Cint}(undef, n)
575589 dev_info = ROCVector{Cint}(undef, 1 )
576590 roc_uplo = convert(rocSOLVER. rocblas_fill, uplo)
577591 $ heevx(dh, jobz, range, roc_uplo, n, A, lda, vl, vu, il, iu, abstol, nev, W, V, ldv, ifail, dev_info)
578592
579593 info = @allowscalar dev_info[1 ]
580594 chkargsok(BlasInt(info))
581- m = @allowscalar nev[1 ]
595+ m = @allowscalar nev[1 ]
582596 return W, V, m
583597 end
584- function heevj!(A:: StridedROCMatrix{$elty} ,
585- W:: StridedROCVector{$relty} ,
586- V:: StridedROCMatrix{$elty} ;
587- uplo:: Char = ' U'
588- tol:: $relty = eps($ relty),
589- max_sweeps:: Int = 100 ,
590- sort:: Char = ' N'
598+ function heevj!(
599+ A:: StridedROCMatrix{$elty} ,
600+ W:: StridedROCVector{$relty} ,
601+ V:: StridedROCMatrix{$elty} ;
602+ uplo:: Char = ' U'
603+ tol:: $relty = eps($ relty),
604+ max_sweeps:: Int = 100 ,
605+ sort:: Char = ' N'
606+ )
591607 chkuplo(uplo)
592608 n = checksquare(A)
593609 lda = max(1 , stride(A, 2 ))
0 commit comments