Fix numpy upgrade to 2.0.0 BUFSIZE import error

microsoft / DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

https://www.deepspeed.ai/

Apache License 2.0

33.6k stars 3.94k forks source link

Fix numpy upgrade to 2.0.0 BUFSIZE import error #5680

Closed Yejing-Lai closed 1 week ago

Yejing-Lai commented 1 week ago

Numpy 2.0.0 removed the BUFSIZE API, replacing it with 'numpy._core.umath.BUFSIZE'.

The minimum numpy version supported by the new code is 1.26.1. If you need to support the lower numpy version, please tell me and we can add another path.

tjruwase commented 1 week ago

@Yejing-Lai, thanks for the PR. Please add support for older numpy versions.

Yejing-Lai commented 1 week ago

@Yejing-Lai, thanks for the PR. Please add support for older numpy versions.

Added older version support. please kindly review. Thanks~