mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-02-21 22:43:18 +00:00
41 lines
1.3 KiB
YAML
41 lines
1.3 KiB
YAML
name: Build
|
|
on:
|
|
pull_request:
|
|
types: [review_requested, synchronize]
|
|
branches:
|
|
- "*"
|
|
|
|
jobs:
|
|
build:
|
|
name: Build and test Colossal-AI
|
|
runs-on: [self-hosted, gpu]
|
|
container:
|
|
image: nvcr.io/nvidia/pytorch:21.07-py3
|
|
options: --gpus all --rm --ipc=host -v /data/cifar-10:/data/cifar-10
|
|
timeout-minutes: 1200
|
|
if: github.event.pull_request.draft == false && github.base_ref == 'main' && github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI' && toJson(github.event.pull_request.requested_reviewers) != '[]'
|
|
steps:
|
|
- name: Setup Environment
|
|
run: |
|
|
export https_proxy=http://172.17.0.1:7890 http_proxy=http://172.17.0.1:7890 all_proxy=socks5://172.17.0.1:7890
|
|
- name: Install dependencies
|
|
run: |
|
|
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
|
|
pip install -U pip setuptools wheel --user
|
|
pip install pytest tensorboard deepspeed apex
|
|
- uses: actions/checkout@v2
|
|
- name: Install Colossal-AI
|
|
run: |
|
|
pip install -v --no-cache-dir --global-option="--cuda_ext" .
|
|
- name: Unit Testing
|
|
run: |
|
|
pytest tests
|
|
env:
|
|
DATA: /data/cifar-10
|
|
|
|
|
|
|
|
|
|
|
|
|