mirror of https://github.com/hpcaitech/ColossalAI
[workflow] fixed trigger condition for 8-gpu unit test (#1323)
parent
e8acf55e8b
commit
4d5dbf48a6
|
@ -30,7 +30,7 @@ jobs:
|
||||||
- name: Unit Testing
|
- name: Unit Testing
|
||||||
run: |
|
run: |
|
||||||
gpu_used=$(nvidia-smi -i 0 --query-gpu=memory.used --format=csv,noheader,nounits)
|
gpu_used=$(nvidia-smi -i 0 --query-gpu=memory.used --format=csv,noheader,nounits)
|
||||||
[ "$gpu_used" -gt "100" ] && PYTHONPATH=$PWD pytest tests
|
[ "$gpu_used" -le "100" ] && PYTHONPATH=$PWD pytest tests
|
||||||
env:
|
env:
|
||||||
DATA: /data/scratch/cifar-10
|
DATA: /data/scratch/cifar-10
|
||||||
|
|
Loading…
Reference in New Issue