yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
missing `#include <cstring>`.
This commit is contained in:
John Bachan 2021-09-23 09:52:42 -07:00
parent 4ec992fab7
commit 30ca3fcacf
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/enqueue.cc
View File

@ -10,6 +10,8 @@
#include "gdrwrap.h" #include "gdrwrap.h"
#include "bootstrap.h" #include "bootstrap.h"
#include <cstring> // std::memcpy
// Only generate inline kernels for LL // Only generate inline kernels for LL
#define NCCL_FUNC5(func, algo, devredop, dtype) \ #define NCCL_FUNC5(func, algo, devredop, dtype) \
(void*)NCCL_KERN_NAME(func, algo, LL, devredop, dtype), \ (void*)NCCL_KERN_NAME(func, algo, LL, devredop, dtype), \