Update descriptor TriangularOperator for AMD GPUs

JuliaSmoothOptimizers · Dec 7, 2023 · 2781c1d · 2781c1d
1 parent 9910494
commit 2781c1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/AMDGPU/operators.jl b/ext/AMDGPU/operators.jl
@@ -132,8 +132,8 @@ for (SparseMatrixType, BlasType) in ((:(ROCSparseMatrixCSR{T}), :BlasFloat),
         end
 
         function KP.update!(A::AMD_TriangularOperator{T}, B::$SparseMatrixType) where T <: $BlasFloat
-            descB = rocSPARSE.ROCSparseMatrixDescriptor(B, 'O')
-            A.descA = descB
+            (B isa ROCSparseMatrixCOO) && rocSPARSE.rocsparse_coo_set_pointers(A.descA, B.rowInd, B.colInd, B.nzVal)
+            (B isa ROCSparseMatrixCSR) && rocSPARSE.rocsparse_csr_set_pointers(A.descA, B.rowPtr, B.colVal, B.nzVal)
             return A
         end
     end