Skip to content
This repository was archived by the owner on Oct 19, 2024. It is now read-only.

Commit 28cb518

Browse files
sammeralomairSammer Alomair
andauthored
removed mapping of numpy.int and numpy.float to nccl and gloo types because they're deprecated (#921)
Co-authored-by: Sammer Alomair <[email protected]>
1 parent efe417d commit 28cb518

File tree

2 files changed

+0
-4
lines changed

2 files changed

+0
-4
lines changed

alpa/collective/collective_group/gloo_util.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121

2222
NUMPY_GLOO_DTYPE_MAP = {
2323
# INT types
24-
numpy.int: pygloo.glooDataType_t.glooInt64,
2524
numpy.uint8: pygloo.glooDataType_t.glooUint8,
2625
numpy.uint32: pygloo.glooDataType_t.glooUint32,
2726
numpy.uint64: pygloo.glooDataType_t.glooUint64,
@@ -30,7 +29,6 @@
3029
numpy.int64: pygloo.glooDataType_t.glooInt64,
3130
# FLOAT types
3231
numpy.half: pygloo.glooDataType_t.glooFloat16,
33-
numpy.float: pygloo.glooDataType_t.glooFloat64,
3432
numpy.float16: pygloo.glooDataType_t.glooFloat16,
3533
numpy.float32: pygloo.glooDataType_t.glooFloat32,
3634
numpy.float64: pygloo.glooDataType_t.glooFloat64,

alpa/collective/collective_group/nccl_util.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@
2929
# cupy types are the same with numpy types
3030
NUMPY_NCCL_DTYPE_MAP = {
3131
# INT types
32-
numpy.int: nccl.NCCL_INT64,
3332
numpy.uint8: nccl.NCCL_UINT8,
3433
numpy.uint32: nccl.NCCL_UINT32,
3534
numpy.uint64: nccl.NCCL_UINT64,
@@ -39,7 +38,6 @@
3938
# FLOAT types
4039
numpy.half: nccl.NCCL_HALF,
4140
# note that numpy.float is float64.
42-
numpy.float: nccl.NCCL_FLOAT64,
4341
numpy.float16: nccl.NCCL_FLOAT16,
4442
numpy.float32: nccl.NCCL_FLOAT32,
4543
numpy.float64: nccl.NCCL_FLOAT64,

0 commit comments

Comments
 (0)