Update `ALCF/helpers.sh`

argonne-lcf / Megatron-DeepSpeed

Ongoing research training transformer language models at scale, including: BERT & GPT-2

Other

9 stars 12 forks source link

Update `ALCF/helpers.sh` #8

Closed saforem2 closed 6 months ago

saforem2 commented 6 months ago

Catch:

if [[ $(hostname) == x3* ]]; then
  if [[ "${PBS_O_HOST}" == sirius* ]]; then
      export MACHINE="Sirius"
  else
      export MACHINE="Polaris"
  fi
fi

Be explicit about "${WORKING_DIR}" throughout