Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cupy_nccl.h 198 Bytes
Newer Older
root's avatar
clone code from github  
root committed
1
2
3
4
5
6
7
8
9
10
11
12 
#ifndef INCLUDE_GUARD_CUDA_CUPY_NCCL_H
#define INCLUDE_GUARD_CUDA_CUPY_NCCL_H

#include <nccl.h>

#ifndef NCCL_MAJOR
#ifndef CUDA_HAS_HALF
#define ncclHalf ((ncclDataType_t)2)
#endif
#endif

#endif