search

pytorch / benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
BSD 3-Clause "New" or "Revised" License
817 stars 259 forks source link

Upgrade numpy to 2.0.0rc #2311

Closed xuzhao9 closed 2 weeks ago

xuzhao9 commented 2 weeks ago

PyTorch upgraded to numpy 2.0.0rc1 which breaks a couple of models, we can fix all of them except DALLE2_pytorch.

DALLE2_pytorch requires embeeding-reader, which requires numpy<2: https://github.com/rom1504/embedding-reader/blob/a4fd55830a502685600ed8ef07947cd1cb92b083/requirements.txt#L3

Therefore, we have to move DALLE2_pytorch to canary_models directory.

Pytorch PR: https://github.com/pytorch/pytorch/pull/128879 Pytorch SEV: https://github.com/pytorch/pytorch/issues/128845

facebook-github-bot commented 2 weeks ago

@xuzhao9 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

facebook-github-bot commented 2 weeks ago

@xuzhao9 has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

facebook-github-bot commented 2 weeks ago

@xuzhao9 merged this pull request in pytorch/benchmark@b949580bb3b481d1a18a958a4f258968543e58a3.