DeepSpeed
384e6d34 - fix: DeepCompile for torch 2.8 (#7402)

Commit
184 days ago
fix: DeepCompile for torch 2.8 (#7402) In torch v2.8.0, all symm mem code are moved into a dedicated folder https://github.com/pytorch/pytorch/commit/ffc6cbfaf78ca219092ce64dcf113377ae698300 So this PR tries to address this change by checking if we have it located under `torch/csrc/distributed/c10d/symm_mem/SymmetricMemory.hpp` (new location). If not, we fall back to the original place for backward compatibilities. This PR also clean up some includes in `z1/2/3.cpp` that has already been included in `deepcompile.h` Signed-off-by: Hollow Man <hollowman@opensuse.org>
Author
Parents
Loading