[feature request] Numpy 2 compatibility

Lightning-AI / lightning-thunder

Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.

Apache License 2.0

1.07k stars 60 forks source link

[feature request] Numpy 2 compatibility #618

Open nikitaved opened 1 week ago

nikitaved commented 1 week ago

It seems like it is quite easy to make Thunder support NumPy 2, modulo dependencies (fdm, for example).

A temporary fix could be disabling envs with NumPy 2 in our CIs, https://github.com/Lightning-AI/lightning-thunder/pull/617.

mruberry commented 4 days ago

triage review: let's do it! fdm is no longer supported; let's do what we need to refactor its use in tests or copy/fork the relevant code and update it