Skip to content

Commit 5450d65

Browse files
committed
use sync
Signed-off-by: niranda perera <niranda.perera@gmail.com>
1 parent eafbea9 commit 5450d65

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

cpp/include/rapidsmpf/detail/rmm_resource_adaptor_impl.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -220,6 +220,7 @@ class RmmResourceAdaptorImpl {
220220
std::size_t alignment = rmm::CUDA_ALLOCATION_ALIGNMENT
221221
) noexcept {
222222
deallocate(sync_stream_, ptr, bytes, alignment);
223+
sync_stream_.synchronize_no_throw();
223224
}
224225

225226
/// @brief Tag this resource as device-accessible for the CCCL concept.

0 commit comments

Comments
 (0)