diff options
| author | Marcin Junczys-Dowmunt <marcinjd@microsoft.com> | 2018-12-07 04:44:15 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-12-07 04:44:15 +0300 |
| commit | b66799babd52c35bbd128644347f78162dc2bbc3 (patch) | |
| tree | 4aeb415b314ceefdf016b24a344be1f4f0853e8d | |
| parent | e327494e4df8a715aa3ad9f8bd040f630a50bb7e (diff) | |
Add comment on nccl.h
| -rwxr-xr-x | src/training/communicator_nccl.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/training/communicator_nccl.h b/src/training/communicator_nccl.h index ac6ffc71..16c49bdb 100755 --- a/src/training/communicator_nccl.h +++ b/src/training/communicator_nccl.h @@ -4,6 +4,7 @@ #include "3rd_party/threadpool.h" #include "tensors/gpu/cuda_helpers.h" +// Generated by NCCL make files in build/nccl/include #include "nccl.h" #include <cuda_runtime.h> |