globus-labs / mof-generation-at-scale / 14375045611

48%

push

github

Introduce DDP training for DiffLinker on Aurora (#187)

* updates for ddp training on aurora

* add PBSClusterEnvironment for Aurora

* allow existing checkpoint directory

* update for specific fork of lightning

* update lightning dependency for xpu

* move lightning install to build-aurora.sh

* fix typo

* put changes for aurora install in build-aurora

* Flake8 fixes

* Remove MPI code (temporarily)

* Add hooks for determining rank with Parsl

* Start moving component test to multi-node training

* Launch more than one training rank

* Use a full node for training

* Flake8 fixes

* Fix mpiexec call, attempt to set MASTER_ADDR

---------

Co-authored-by: Azton Wells <azton.wells@gmail.com>

61 of 98 new or added lines in 5 files covered. (62.24%)

4440 of 9797 relevant lines covered (45.32%)

0.45 hits per line