Skip to content

support offline consolidate and reshard fsdp checkpoints #85

support offline consolidate and reshard fsdp checkpoints

support offline consolidate and reshard fsdp checkpoints #85

Workflow file for this run

name: Unit Tests
on:
workflow_dispatch:
pull_request:
branches:
- main
tags:
- r[0-9]+.[0-9]+
paths-ignore:
- 'docs/**'
- '**.md'
push:
branches:
- main
tags:
- r[0-9]+.[0-9]+
paths-ignore:
- 'docs/**'
- '**.md'
jobs:
run-shell-script:
runs-on: self-hosted
steps:
- name: Checkout code
uses: actions/checkout@v3
- name: Run unit test
run: |
docker pull $UT_IMAGE
docker run -v $PWD:$PWD -w $PWD --net host --ipc host --shm-size 80G -t --rm --gpus all $UT_IMAGE bash -c '
pip install -r requirements/requirements-test.txt && \
git config --global --add safe.directory $PWD && \
pip install -e . && \
bash tests/run_ut.sh'
env:
UT_IMAGE: ${{ secrets.UT_IMAGE }}