Fix the error with Hermtian matrices

exanauts · Dec 8, 2023 · 59aed24 · 59aed24
1 parent 006e180
commit 59aed24
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/generic.jl b/src/generic.jl
@@ -12,6 +12,7 @@ function LinearAlgebra.ldlt(A::CuSparseMatrixCSR{T,Cint}) where T <: BlasFloat
   n = LinearAlgebra.checksquare(A)
   structure = T <: Real ? 'S' : 'H'
   solver = CudssSolver(A, structure, 'F')
+  (T <: Complex) && cudss_set(solver, "pivot_type", 'N')
   x = CudssMatrix(T, n)
   b = CudssMatrix(T, n)
   cudss("analysis", solver, x, b)

diff --git a/test/test_cudss.jl b/test/test_cudss.jl
@@ -230,7 +230,6 @@ function cudss_generic()
         B_gpu = CuMatrix(B_cpu)
 
         solver = ldlt(A_gpu)
-        (T <: Complex) && cudss_set(solver, "pivot_type", 'N')
         ldiv!(X_gpu, solver, B_gpu)
         R_gpu = B_gpu - A_gpu * X_gpu
         @test norm(R_gpu) ≤ √eps(R)