DeepSpeed
04d69cc6 - Fix import deepspeed crash on PyTorch v2.3 + Python 3.12 (#7875)

Commit
9 days ago
Fix import deepspeed crash on PyTorch v2.3 + Python 3.12 (#7875) `import deepspeed` raises `RuntimeError: Dynamo is not supported on Python 3.12+` on PyTorch 2.3 + Python 3.12. The `jit_script_compat` decorator (introduced in #7835) calls `torch.compile()` unconditionally on PyTorch >= 2.0, but Dynamo support for Python 3.12 was only added in PyTorch 2.4. Multiple eager import chains trigger this decorator at import time, crashing before user code runs. This PR adds a version gate to skip torch.compile on known-unsupported combinations, plus a double fallback (torch.compile → torch.jit.script → identity function) so the decorator won't crash. --------- Signed-off-by: Masahiro Tanaka <mtanaka@anyscale.com> Co-authored-by: Olatunji Ruwase <tjruwase@gmail.com>
Author
Parents
Loading