DeepSpeed
6d7b44a8 - [NPU] load EXPORT_ENV based on different accelerators to support multi-node training on other devices (#4830)

Commit
2 years ago
[NPU] load EXPORT_ENV based on different accelerators to support multi-node training on other devices (#4830) Different hardwares may require different environment variables. To support multi-node training feature for NPU and other devices that rely on different env vars, I add a method `export_envs()` to each accelerator and load them in runner.py For works about NPU, see #4567 Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
Parents
Loading