DeepSpeed
fcc731f0 - Fix torch.compile error for PyTorch v2.3 (#5463)

Commit
1 year ago
Fix torch.compile error for PyTorch v2.3 (#5463) PyTorch v2.3 throws an error when it tries to compile `iter_params` used for ZeRO3. This PR excludes the function from the compilation targets. After this PR is merged, we can [unpin the torch version for unit tests](https://github.com/microsoft/DeepSpeed/pull/5459).
Author
Parents
Loading