Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
kmp5VT committed Nov 1, 2023
1 parent 48a8c15 commit 02cd879
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions NDTensors/ext/NDTensorsCUDAExt/linearalgebra.jl
Original file line number Diff line number Diff line change
Expand Up @@ -20,20 +20,20 @@ function NDTensors.svd_catch_error(A::CuMatrix, ::CUDA.CUSOLVER.QRAlgorithm)
s = size(A)
if s[1] < s[2]
At = copy(Adjoint(A))
USV = try

USV = try
svd(At; alg=CUDA.CUSOLVER.QRAlgorithm())
catch
return nothing
end
MV ,MS, MU = USV;
MV, MS, MU = USV
USV = SVD(copy(MU), MS, Adjoint(MV))
else
USV = try
USV = try
svd(A; alg=CUDA.CUSOLVER.QRAlgorithm())
catch
return nothing
end
end
return USV
end
end

0 comments on commit 02cd879

Please sign in to comment.