Skip to content

Commit fbe6249

Browse files
committed
Use C++ header instead of CUDA header
1 parent 56246c6 commit fbe6249

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

include/cuco/detail/utility/memcpy_async.cuh renamed to include/cuco/detail/utility/memcpy_async.hpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@
2020

2121
#include <cuda/stream_ref>
2222

23+
#include <cstddef>
24+
2325
namespace cuco::detail {
2426

2527
/**
@@ -41,7 +43,7 @@ namespace cuco::detail {
4143
if (dst == nullptr || src == nullptr || count == 0) { return cudaSuccess; }
4244

4345
#if CUDART_VERSION >= 12080
44-
if (stream.get() == 0) { return cudaMemcpyAsync(dst, src, count, kind, stream.get()); }
46+
if (stream.get() == nullptr) { return cudaMemcpyAsync(dst, src, count, kind, stream.get()); }
4547

4648
void* dsts[1] = {dst};
4749
void* srcs[1] = {const_cast<void*>(src)};
@@ -63,4 +65,5 @@ namespace cuco::detail {
6365
return cudaMemcpyAsync(dst, src, count, kind, stream.get());
6466
#endif // CUDART_VERSION >= 12080
6567
}
68+
6669
} // namespace cuco::detail

0 commit comments

Comments
 (0)