Update include/cuco/detail/storage/counter_storage.cuh

Co-authored-by: Bradley Dice <[email protected]>

Author: PointKernel

include/cuco/detail/storage/counter_storage.cuh

@@ -96,7 +96,11 @@ class counter_storage : public storage_base<cuco::extent<SizeType, 1>> {
     CUCO_CUDA_TRY(cudaMallocHost(&h_count, sizeof(size_type)));
     CUCO_CUDA_TRY(cudaMemcpyAsync(
       h_count, this->data(), sizeof(size_type), cudaMemcpyDeviceToHost, stream.get()));
+#if CCCL_MAJOR_VERSION > 3 || (CCCL_MAJOR_VERSION == 3 && CCCL_MINOR_VERSION >= 1)
     stream.sync();
+#else   // ^^^ CCCL >= 3.1 ^^^ / vvv CCCL < 3.1 vvv
+    stream.wait();
+#endif  // CCCL < 3.1
     size_type result = *h_count;
     CUCO_CUDA_TRY(cudaFreeHost(h_count));
     return result;