Unverified Commit 76edb04b authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[workflow] adjust the GPU memory threshold for scheduled unit test (#2558)

* [workflow] adjust the GPU memory threshold for scheduled unit test

* polish code
parent ba475173
name: Build on 8 GPUs
name: Build on Schedule
on:
schedule:
......@@ -39,7 +39,7 @@ jobs:
- name: Unit Testing
run: |
gpu_used=$(nvidia-smi -i 0 --query-gpu=memory.used --format=csv,noheader,nounits)
[ "$gpu_used" -le "100" ] && PYTHONPATH=$PWD pytest tests
[ "$gpu_used" -le "10000" ] && PYTHONPATH=$PWD pytest tests
env:
DATA: /data/scratch/cifar-10
LD_LIBRARY_PATH: /github/home/.tensornvme/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment