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 name: Build on PR
on: on:
pull_request: pull_request:
......
name: Build on 8 GPUs name: Build on Schedule
on: on:
schedule: schedule:
...@@ -39,7 +39,7 @@ jobs: ...@@ -39,7 +39,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" -le "100" ] && PYTHONPATH=$PWD pytest tests [ "$gpu_used" -le "10000" ] && PYTHONPATH=$PWD pytest tests
env: env:
DATA: /data/scratch/cifar-10 DATA: /data/scratch/cifar-10
LD_LIBRARY_PATH: /github/home/.tensornvme/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 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