mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
576e0896b7
* feat(runtime): add support for launcher resource allocation in MPI jobs Signed-off-by: Andrey Velichkevich <[email protected]> * Add unit tests Signed-off-by: Andrey Velichkevich <[email protected]> * Set numProcPerNode for MPI plugin Signed-off-by: Andrey Velichkevich <[email protected]> * Move util func to runtime package Signed-off-by: Andrey Velichkevich <[email protected]> * Fix torchtune plugin Signed-off-by: Andrey Velichkevich <[email protected]> * Inline if for GPU check Signed-off-by: Andrey Velichkevich <[email protected]> * Assign container resources once Signed-off-by: Andrey Velichkevich <[email protected]> * Add todo for test wrappers Signed-off-by: Andrey Velichkevich <[email protected]> --------- Signed-off-by: Andrey Velichkevich <[email protected]> Co-authored-by: Andrey Velichkevich <[email protected]>