Skip to content

Commit

Permalink
Fix rocSPARSE bug causing progress issues
Browse files Browse the repository at this point in the history
Issues:

o Ellpack progress tests were failing for rocSPARSE build
o Progress benchmark was also failing for N>=6000

Fix:

o bml_sort_rocsparse_ellpack was obtaining nnz from host array
o changed this to obtain nnz from device array
  • Loading branch information
mewall committed Jul 3, 2023
1 parent 5aa44ec commit a688b4e
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion src/C-interface/ellpack/bml_allocate_ellpack_typed.c
Original file line number Diff line number Diff line change
Expand Up @@ -441,7 +441,10 @@ void TYPED_FUNC(
size_t lworkInBytes = 0;
char *dwork = NULL;

nnz = csrRowPtr[N];
#pragma omp target map(to:N) map(from:nnz)
{
nnz = csrRowPtr[N];
}
// Temporary array for sorting

csrVal_tmp = (REAL_T *) malloc(sizeof(REAL_T)* nnz);
Expand Down

0 comments on commit a688b4e

Please sign in to comment.