formatting recommendations

Signed-off-by: Steven Hahn <[email protected]>

Author: quantumsteve

lib/cusolver/dense_generic.jl

@@ -505,13 +505,13 @@ function Xgeev!(jobvl::Char, jobvr::Char, A::StridedCuMatrix{T}) where {T <: Bla
 end
 
 # XsyevBatched
-function XsyevBatched!(jobz::Char, uplo::Char, A::StridedCuArray{T,3}) where {T <: BlasFloat}
+function XsyevBatched!(jobz::Char, uplo::Char, A::StridedCuArray{T, 3}) where {T <: BlasFloat}
     CUSOLVER.version() < v"11.7.1" && throw(ErrorException("This operation is not supported by the current CUDA version."))
     chkuplo(uplo)
     n = checksquare(A)
-    batch_size = size(A,3)
+    batch_size = size(A, 3)
     R = real(T)
-    lda = max(1, stride(A,2))
+    lda = max(1, stride(A, 2))
     W = CuMatrix{R}(undef, n, batch_size)
     params = CuSolverParameters()
     dh = dense_handle()
@@ -520,18 +520,22 @@ function XsyevBatched!(jobz::Char, uplo::Char, A::StridedCuArray{T,3}) where {T
     function bufferSize()
         out_cpu = Ref{Csize_t}(0)
         out_gpu = Ref{Csize_t}(0)
-        cusolverDnXsyevBatched_bufferSize(dh, params, jobz, uplo, n,
-                                          T, A, lda, R, W, T, out_gpu, out_cpu, batch_size)
-        out_gpu[], out_cpu[]
+        cusolverDnXsyevBatched_bufferSize(
+            dh, params, jobz, uplo, n,
+            T, A, lda, R, W, T, out_gpu, out_cpu, batch_size
+        )
+        return out_gpu[], out_cpu[]
     end
     with_workspaces(dh.workspace_gpu, dh.workspace_cpu, bufferSize()...) do buffer_gpu, buffer_cpu
-        cusolverDnXsyevBatched(dh, params, jobz, uplo, n, T, A,
-                               lda, R, W, T, buffer_gpu, sizeof(buffer_gpu),
-                               buffer_cpu, sizeof(buffer_cpu), dh.info, batch_size)
+        cusolverDnXsyevBatched(
+            dh, params, jobz, uplo, n, T, A,
+            lda, R, W, T, buffer_gpu, sizeof(buffer_gpu),
+            buffer_cpu, sizeof(buffer_cpu), dh.info, batch_size
+        )
     end
 
     info = @allowscalar collect(dh.info)
-    for i = 1:batch_size
+    for i in 1:batch_size
         chkargsok(info[i] |> BlasInt)
     end
 

test/libraries/cusolver/dense_generic.jl

@@ -39,21 +39,21 @@ p = 5
 
                 A = rand(elty, n, n, batch_size)
                 B = rand(elty, n, n, batch_size)
-                for i = 1:batch_size
-                    S = rand(elty,n,n)
+                for i in 1:batch_size
+                    S = rand(elty, n, n)
                     S = S * S' + I
-                    B[:,:,i] .= S
+                    B[:, :, i] .= S
                     S = uplo == 'L' ? tril(S) : triu(S)
-                    A[:,:,i] .= S
+                    A[:, :, i] .= S
                 end
                 d_A = CuArray(A)
                 d_W, d_V = CUSOLVER.XsyevBatched!('V', uplo, d_A)
                 W = collect(d_W)
                 V = collect(d_V)
-                for i = 1:batch_size
-                    Bᵢ = B[:,:,i]
-                    Wᵢ = Diagonal(W[:,i])
-                    Vᵢ = V[:,:,i]
+                for i in 1:batch_size
+                    Bᵢ = B[:, :, i]
+                    Wᵢ = Diagonal(W[:, i])
+                    Vᵢ = V[:, :, i]
                     @test Bᵢ * Vᵢ ≈ Vᵢ * Diagonal(Wᵢ)
                 end