diff options
Diffstat (limited to 'src/collectives/device/all_gather.cu')
| -rw-r--r-- | src/collectives/device/all_gather.cu | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/src/collectives/device/all_gather.cu b/src/collectives/device/all_gather.cu index 0f572ce..530bf14 100644 --- a/src/collectives/device/all_gather.cu +++ b/src/collectives/device/all_gather.cu @@ -4,12 +4,8 @@ * See LICENSE.txt for license information ************************************************************************/ -#include "common.h" #include "all_gather.h" +#include "common.h" #include "collectives.h" -#define UNROLL 4 - -#if NCCL_OP == 0 -IMPL_COLL3(ncclAllGather, copy, FuncSum, i8, int8_t, ncclCollAllGather, ncclSum, ncclInt8); -#endif +IMPL_COLL_C(ncclAllGather, ncclCollAllGather); |