diff options
| -rw-r--r-- | src/enqueue.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/enqueue.cc b/src/enqueue.cc index e863b82..4deac18 100644 --- a/src/enqueue.cc +++ b/src/enqueue.cc @@ -10,6 +10,8 @@ #include "gdrwrap.h" #include "bootstrap.h" +#include <cstring> // std::memcpy + // Only generate inline kernels for LL #define NCCL_FUNC5(func, algo, devredop, dtype) \ (void*)NCCL_KERN_NAME(func, algo, LL, devredop, dtype), \ |