Skip to content

Commit

Permalink
Fix an error for TriangularOperator on AMD GPUs
Browse files Browse the repository at this point in the history
  • Loading branch information
amontoison committed Dec 5, 2023
1 parent 136292e commit 829659f
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions ext/AMDGPU/operators.jl
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,10 @@ for (SparseMatrixType, BlasType) in ((:(ROCSparseMatrixCSR{T}), :BlasFloat),
m,n = size(A)
alpha = Ref{T}(one(T))
descA = rocSPARSE.ROCSparseMatrixDescriptor(A, 'O')
rocsparse_uplo = Ref{rocSPARSE.rocsparse_diag_type}(uplo)
rocsparse_diag = Ref{rocSPARSE.rocsparse_matrix_type}(diag)
rocSPARSE.rocsparse_spmat_set_attribute(descA, rocSPARSE.rocsparse_spmat_fill_mode, rocsparse_uplo, Csize_t(sizeof(rocsparse_uplo)))
rocSPARSE.rocsparse_spmat_set_attribute(descA, rocSPARSE.rocsparse_spmat_diag_type, rocsparse_diag, Csize_t(sizeof(rocsparse_diag)))
if nrhs == 1
descX = rocSPARSE.ROCDenseVectorDescriptor(T, n)
descY = rocSPARSE.ROCDenseVectorDescriptor(T, m)
Expand Down

0 comments on commit 829659f

Please sign in to comment.