2024-11-21 nightly release (332760d4b300f00a0d862e3cfe1495db3b1a14f9) #2082
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Unittests on Windows GPU | |
on: | |
pull_request: | |
push: | |
branches: | |
- nightly | |
- main | |
- release/* | |
workflow_dispatch: | |
jobs: | |
unittests-windows-gpu: | |
uses: pytorch/test-infra/.github/workflows/windows_job.yml@main | |
with: | |
repository: pytorch/audio | |
runner: windows.g5.4xlarge.nvidia.gpu | |
timeout: 360 | |
script: | | |
# Mark Build Directory Safe | |
git config --global --add safe.directory /__w/audio/audio | |
# Set up Environment Variables | |
export PYTHON_VERSION="3.8" | |
export CUDA_VERSION=11.8 | |
# Set CHANNEL | |
if [[(${GITHUB_EVENT_NAME} = 'pull_request' && (${GITHUB_BASE_REF} = 'release'*)) || (${GITHUB_REF} = 'refs/heads/release'*) ]]; then | |
export UPLOAD_CHANNEL=test | |
else | |
export UPLOAD_CHANNEL=nightly | |
fi | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_APPLY_CMVN_SLIDING=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_FBANK_FEATS=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_KALDI_PITCH_FEATS=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_MFCC_FEATS=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_COMPUTE_SPECTROGRAM_FEATS=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_CMD_SOX=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_KALDI=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_SOX=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_ON_PYTHON_310=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MOD_sentencepiece=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_AUDIO_OUT_DEVICE=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_MACOS=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_CUDA_SMALL_MEMORY=true | |
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_TEMPORARY_DISABLED=true | |
.github/scripts/unittest-windows/setup_env.sh | |
./packaging/windows/internal/cuda_install.bat | |
./packaging/windows/internal/driver_update.bat | |
.github/scripts/unittest-windows/install.sh | |
.github/scripts/unittest-windows/run_test.sh |