fix compile rocsparseWellContrib

This commit is contained in:
Razvan Nane 2024-05-31 11:55:53 +02:00
parent c92d80e0e8
commit 988d3821cd

View File

@ -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));