/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
.github/workflows/run_chatgpt_unit_tests.yml
52 строки
1 KB
YeAnbang
Merge branch 'grpo-latest-rebase-main' of https://github.com/hpcaitech/ColossalAI into grpo-latest-rebase-main
14 авг 2025, 14:03
14 авг 2025, 14:03
99ba48f
Код
Авторство
О чём код?
name: Run ChatGPT unit tests on: pull_request: types: [synchronize, opened, reopened] paths: - 'applications/ColossalChat/coati/**' - 'applications/ColossalChat/requirements.txt' - 'applications/ColossalChat/setup.py' - 'applications/ColossalChat/tests/**' - 'applications/ColossalChat/pytest.ini' jobs: tests: name: Run ChatGPT unit tests if: | github.event.pull_request.draft == false && github.base_ref == 'main' && github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI' runs-on: [self-hosted, ubuntu-latest] container: image: image-cloud.luchentech.com/hpcaitech/pytorch-cuda:2.2.2-12.1.0 options: --gpus all --rm -v /data/scratch/examples-data:/data/scratch/examples-data timeout-minutes: 180 defaults: run: shell: bash steps: - name: Checkout ColossalAI uses: actions/checkout@v2 - name: Install ChatGPT env: CFLAGS: "-O1" CXXFLAGS: "-O1" MAX_JOBS: 4 run: | pip install flash-attn --no-build-isolation cd applications/ColossalChat pip install -v . pip install pytest - name: Execute Unit Testing run: | cd applications/ColossalChat rm -rf ~/.cache/colossalai pytest tests/ cd ./tests ./test_templating.sh env: NCCL_SHM_DISABLE: 1 MAX_JOBS: 8