Specialize cholcopy to avoid scalar indexing. (#1716)

JuliaGPU · Jan 5, 2023 · f9af125 · f9af125
1 parent 94e7edb
commit f9af125
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/cusolver/linalg.jl b/lib/cusolver/linalg.jl
@@ -356,4 +356,7 @@ if VERSION >= v"1.8-"
         C, info = LinearAlgebra._chol!(copy(parent(A)), A.uplo == 'U' ? UpperTriangular : LowerTriangular)
         return Cholesky(C.data, A.uplo, info)
     end
+
+    LinearAlgebra.cholcopy(A::LinearAlgebra.RealHermSymComplexHerm{<:Any,<:CuArray}) =
+        copyto!(similar(A, LinearAlgebra.choltype(A)), A)
 end