diff options
Diffstat (limited to 'src/collectives/device/broadcast.cu')
-rw-r--r-- | src/collectives/device/broadcast.cu | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/src/collectives/device/broadcast.cu b/src/collectives/device/broadcast.cu new file mode 100644 index 0000000..4125de4 --- /dev/null +++ b/src/collectives/device/broadcast.cu @@ -0,0 +1,15 @@ +/************************************************************************* + * Copyright (c) 2015-2018, NVIDIA CORPORATION. All rights reserved. + * + * See LICENSE.txt for license information + ************************************************************************/ + +#include "common.h" +#include "broadcast.h" +#include "collectives.h" + +#define UNROLL 4 + +#if NCCL_OP == 0 +IMPL_COLL3(ncclBroadcast, copy, FuncSum, i8, int8_t, ncclCollBroadcast, ncclSum, ncclInt8); +#endif |