DeepSpeed
3255569b - Switch hasattr check from compile to compiler (#5096)

Commit
1 year ago
Switch hasattr check from compile to compiler (#5096) torch.compile introduced in torch 2.0 but torch.compiler was introduced in torch 2.1, this fixes issues for those building with torch 2.0 like the A6000 builds. ``` >>> torch.__version__ '2.1.0+cu121' >>> hasattr(torch, "compile") True >>> hasattr(torch, "compiler") True >>> torch.__version__ '2.0.0+cu117' >>> hasattr(torch, "compile") True >>> hasattr(torch, "compiler") False ```
Author
Parents
Loading