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
|
||||
run: |
|
||||
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:
|
||||
DATA: /data/scratch/cifar-10
|
||||
|
Loading…
Reference in New Issue