From 76edb04b0d79860358cf844fbcbecc81a84c14ed Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Mon, 6 Feb 2023 13:47:25 +0800 Subject: [PATCH] [workflow] adjust the GPU memory threshold for scheduled unit test (#2558) * [workflow] adjust the GPU memory threshold for scheduled unit test * polish code --- .github/workflows/{build.yml => build_on_pr.yml} | 2 +- .github/workflows/{build_gpu_8.yml => build_on_schedule.yml} | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) rename .github/workflows/{build.yml => build_on_pr.yml} (99%) rename .github/workflows/{build_gpu_8.yml => build_on_schedule.yml} (94%) diff --git a/.github/workflows/build.yml b/.github/workflows/build_on_pr.yml similarity index 99% rename from .github/workflows/build.yml rename to .github/workflows/build_on_pr.yml index 3c163e774..82b671ace 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build_on_pr.yml @@ -1,4 +1,4 @@ -name: Build +name: Build on PR on: pull_request: diff --git a/.github/workflows/build_gpu_8.yml b/.github/workflows/build_on_schedule.yml similarity index 94% rename from .github/workflows/build_gpu_8.yml rename to .github/workflows/build_on_schedule.yml index be8337dd0..ea1f4879c 100644 --- a/.github/workflows/build_gpu_8.yml +++ b/.github/workflows/build_on_schedule.yml @@ -1,4 +1,4 @@ -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