Fix compilation failure in "src/enqueue.cc" on older GCC because of

missing `#include <cstring>`.
2021-09-23 09:52:42 -07:00 · 2021-09-23 09:52:42 -07:00 · 30ca3fcacf
commit 30ca3fcacf
parent 4ec992fab7
1 changed files with 2 additions and 0 deletions
--- 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), \