Merge pull request #5400 from BigDataAccelerate/fix_rocsparsewells

fix compile rocsparseWellContributions
This commit is contained in:
Arne Morten Kvarving
2024-05-31 13:50:11 +02:00
committed by GitHub
@@ -144,9 +144,9 @@ apply_stdwells([[maybe_unused]] Scalar* d_x,
[[maybe_unused]] Scalar* d_y)
{
#ifdef __HIP__
unsigned gridDim = num_std_wells;
unsigned gridDim = this->num_std_wells;
unsigned blockDim = 64;
unsigned shared_mem_size = (blockDim + 2 * dim_wells) * sizeof(Scalar); // shared memory for localSum, z1 and z2
unsigned shared_mem_size = (blockDim + 2 * this->dim_wells) * sizeof(Scalar); // shared memory for localSum, z1 and z2
// dim3(N) will create a vector {N, 1, 1}
stdwell_apply<<<dim3(gridDim), dim3(blockDim), shared_mem_size, stream>>>(d_Cnnzs_hip,
d_Dnnzs_hip,
@@ -155,8 +155,8 @@ apply_stdwells([[maybe_unused]] Scalar* d_x,
d_Bcols_hip,
d_x,
d_y,
dim,
dim_wells,
this->dim,
this->dim_wells,
d_val_pointers_hip
);
HIP_CHECK(hipStreamSynchronize(stream));