Add some CUSPARSE.

JuliaSmoothOptimizers · Nov 30, 2023 · f2d9f25 · f2d9f25
1 parent c7a662c
commit f2d9f25
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ext/CUDA/ic0.jl b/ext/CUDA/ic0.jl
@@ -34,7 +34,7 @@ for (bname, aname, sname, T) in ((:cusparseScsric02_bufferSize, :cusparseScsric0
       buffer = CuVector{UInt8}(undef, buffer_size[])
       CUSPARSE.$aname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.rowPtr, P.colVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       posit = Ref{Cint}(1)
-      CUSPARSE.cusparseXcsric02_zeroPivot(handle(), info, posit)
+      CUSPARSE.cusparseXcsric02_zeroPivot(CUSPARSE.handle(), info, posit)
       (posit[] ≥ 0) && error("Structural/numerical zero in A at ($(posit[]),$(posit[])))")
       CUSPARSE.$sname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.rowPtr, P.colVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       return NVIDIA_IC0(desc, buffer, info, P)
@@ -50,7 +50,7 @@ for (bname, aname, sname, T) in ((:cusparseScsric02_bufferSize, :cusparseScsric0
       buffer = CuVector{UInt8}(undef, buffer_size[])
       CUSPARSE.$aname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.colPtr, P.rowVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       posit = Ref{Cint}(1)
-      CUSPARSE.cusparseXcsric02_zeroPivot(handle(), info, posit)
+      CUSPARSE.cusparseXcsric02_zeroPivot(CUSPARSE.handle(), info, posit)
       (posit[] ≥ 0) && error("Structural/numerical zero in A at ($(posit[]),$(posit[])))")
       CUSPARSE.$sname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.colPtr, P.rowVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       return NVIDIA_IC0(desc, buffer, info, P)

diff --git a/ext/CUDA/ilu0.jl b/ext/CUDA/ilu0.jl
@@ -34,7 +34,7 @@ for (bname, aname, sname, T) in ((:cusparseScsrilu02_bufferSize, :cusparseScsril
       buffer = CuVector{UInt8}(undef, buffer_size[])
       CUSPARSE.$aname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.rowPtr, P.colVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       posit = Ref{Cint}(1)
-      CUSPARSE.cusparseXcsric02_zeroPivot(handle(), info, posit)
+      CUSPARSE.cusparseXcsric02_zeroPivot(CUSPARSE.handle(), info, posit)
       (posit[] ≥ 0) && error("Structural/numerical zero in A at ($(posit[]),$(posit[])))")
       CUSPARSE.$sname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.rowPtr, P.colVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       return NVIDIA_ILU0(desc, buffer, info, P)
@@ -50,7 +50,7 @@ for (bname, aname, sname, T) in ((:cusparseScsrilu02_bufferSize, :cusparseScsril
       buffer = CuVector{UInt8}(undef, buffer_size[])
       CUSPARSE.$aname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.colPtr, P.rowVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       posit = Ref{Cint}(1)
-      CUSPARSE.cusparseXcsric02_zeroPivot(handle(), info, posit)
+      CUSPARSE.cusparseXcsric02_zeroPivot(CUSPARSE.handle(), info, posit)
       (posit[] ≥ 0) && error("Structural/numerical zero in A at ($(posit[]),$(posit[])))")
       CUSPARSE.$sname(CUSPARSE.handle(), n, nnz(P), desc, P.nzVal, P.colPtr, P.rowVal, info, CUSPARSE.CUSPARSE_SOLVE_POLICY_USE_LEVEL, buffer)
       return NVIDIA_ILU0(desc, buffer, info, P)