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