Reduce Unit Test Times (Part 1) (#3829)

* move torch19 tests to nightly

* make megatron apex install persistent on blob storage
This commit is contained in:
Michael Wyatt
2023-06-28 11:29:47 -07:00
committed by GitHub
parent 25e500e8dd
commit 7726fc8d54
3 changed files with 10 additions and 9 deletions

View File

@ -35,8 +35,17 @@ jobs:
- name: Install apex
run: |
git clone https://github.com/NVIDIA/apex.git
cd apex
# Install 82ee367 specifically before pyproject.toml changes due to incomplete dependencies causing errors.
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" git+https://github.com/NVIDIA/apex.git@82ee367
git checkout 82ee367
CURRENT_VER=$(git rev-parse HEAD)
INSTALLED_VER=$(cat /blob/apex/.venv_installed_version)
if [[ "$CURRENT_VER" != "$INSTALLED_VER" ]]; then
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --target=/blob/apex/ --upgrade .
git rev-parse HEAD > /blob/apex/.venv_installed_version
fi
echo PYTHONPATH=$PYTHONPATH:/blob/apex/ >> $GITHUB_ENV
- name: Python environment
run: |

View File

@ -1,10 +1,6 @@
name: nv-torch19-p40
on:
pull_request:
paths-ignore:
- 'docs/**'
- 'blogs/**'
schedule:
- cron: "0 0 * * *"

View File

@ -1,10 +1,6 @@
name: nv-torch19-v100
on:
pull_request:
paths-ignore:
- 'docs/**'
- 'blogs/**'
schedule:
- cron: "0 0 * * *"