diff --git a/cpp/include/raft/lap/detail/lap_functions.cuh b/cpp/include/raft/lap/detail/lap_functions.cuh index 3a801ff060..e03185f508 100644 --- a/cpp/include/raft/lap/detail/lap_functions.cuh +++ b/cpp/include/raft/lap/detail/lap_functions.cuh @@ -466,7 +466,7 @@ inline void dualUpdate(raft::handle_t const& handle, dim3 threads_per_block; int total_blocks; - rmm::device_scalar sp_min_v(handle.get_stream()); + rmm::device_uvector sp_min_v(SP, handle.get_stream()); raft::lap::detail::calculateLinearDims(blocks_per_grid, threads_per_block, total_blocks, SP); kernel_dualUpdate_1<<>>(