fix CI machine tag

This commit is contained in:
BurkeHulk
2025-06-02 17:45:40 +08:00
parent fd56b22278
commit c9cba49ab5
19 changed files with 24 additions and 20 deletions

View File

@@ -17,7 +17,7 @@ jobs:
github.event.pull_request.draft == false &&
github.base_ref == 'main' &&
github.event.pull_request.base.repo.full_name == 'hpcaitech/ColossalAI'
runs-on: [self-hosted, gpu]
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 --shm-size=10.24gb