diff --git a/src/gpuarray_collectives_cuda_nccl.c b/src/gpuarray_collectives_cuda_nccl.c index dc5a901ab7..44c1297aa7 100644 --- a/src/gpuarray_collectives_cuda_nccl.c +++ b/src/gpuarray_collectives_cuda_nccl.c @@ -175,11 +175,12 @@ static inline ncclDataType_t convert_data_type(int typecode) { switch (typecode) { case GA_BYTE: return ncclChar; case GA_INT: return ncclInt; - case GA_HALF: return ncclHalf; case GA_FLOAT: return ncclFloat; case GA_DOUBLE: return ncclDouble; case GA_LONG: return ncclInt64; case GA_ULONG: return ncclUint64; + case GA_HALF: return ncclHalf; + case GA_FLOAT16: return ncclHalf; } return nccl_NUM_TYPES; }