Skip to content

Commit 2237b39

Browse files
committed
fix cuda ci_image and runner_label
1 parent 0c303bf commit 2237b39

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

.github/configs/cuda.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ hardware_name: cuda
66
display_name: 'CUDA Tests'
77

88
# CI image for BAAI env
9-
# ci_image: harbor.baai.ac.cn/flagscale/megatron-lm-fl:cuda12.8.1-cudnn9.8.0-torch2.7.0-python3.12-dev-02111650
9+
ci_image: harbor.baai.ac.cn/flagscale/megatron-lm-fl:cuda12.8.1-cudnn9.8.0-torch2.7.0-python3.12-dev-02111650
1010

1111
# Runner labels for self-hosted A100 node
1212
# runner_labels:
@@ -18,7 +18,7 @@ display_name: 'CUDA Tests'
1818

1919
# Runner labels for BAAI env
2020
runner_labels:
21-
- nv-8g-cicd-te
21+
- nv-8g-cicd-megatron
2222

2323
# Container volumes (hardware-specific paths)
2424
# container_volumes:

0 commit comments

Comments
 (0)