This repository was archived by the owner on Oct 19, 2024. It is now read-only.
File tree Expand file tree Collapse file tree 2 files changed +0
-4
lines changed
alpa/collective/collective_group Expand file tree Collapse file tree 2 files changed +0
-4
lines changed Original file line number Diff line number Diff line change 21
21
22
22
NUMPY_GLOO_DTYPE_MAP = {
23
23
# INT types
24
- numpy .int : pygloo .glooDataType_t .glooInt64 ,
25
24
numpy .uint8 : pygloo .glooDataType_t .glooUint8 ,
26
25
numpy .uint32 : pygloo .glooDataType_t .glooUint32 ,
27
26
numpy .uint64 : pygloo .glooDataType_t .glooUint64 ,
30
29
numpy .int64 : pygloo .glooDataType_t .glooInt64 ,
31
30
# FLOAT types
32
31
numpy .half : pygloo .glooDataType_t .glooFloat16 ,
33
- numpy .float : pygloo .glooDataType_t .glooFloat64 ,
34
32
numpy .float16 : pygloo .glooDataType_t .glooFloat16 ,
35
33
numpy .float32 : pygloo .glooDataType_t .glooFloat32 ,
36
34
numpy .float64 : pygloo .glooDataType_t .glooFloat64 ,
Original file line number Diff line number Diff line change 29
29
# cupy types are the same with numpy types
30
30
NUMPY_NCCL_DTYPE_MAP = {
31
31
# INT types
32
- numpy .int : nccl .NCCL_INT64 ,
33
32
numpy .uint8 : nccl .NCCL_UINT8 ,
34
33
numpy .uint32 : nccl .NCCL_UINT32 ,
35
34
numpy .uint64 : nccl .NCCL_UINT64 ,
39
38
# FLOAT types
40
39
numpy .half : nccl .NCCL_HALF ,
41
40
# note that numpy.float is float64.
42
- numpy .float : nccl .NCCL_FLOAT64 ,
43
41
numpy .float16 : nccl .NCCL_FLOAT16 ,
44
42
numpy .float32 : nccl .NCCL_FLOAT32 ,
45
43
numpy .float64 : nccl .NCCL_FLOAT64 ,
You can’t perform that action at this time.
0 commit comments