Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions .github/scripts/unittest-windows/install.sh
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,8 @@ fi
printf "Installing PyTorch\n"
pip install --pre torch --index-url https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${wheel}

python -c "import torch; print(torch.__version__)"

torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())")
echo torch.cuda.is_available is $torch_cuda

Expand All @@ -42,6 +44,12 @@ if [ ! -z "${CUDA_VERSION:-}" ] ; then
fi
fi

printf "Installing TorchCodec\n"
# torchcodec nightly has no Windows+CUDA wheels, so we'll use CPU-only
# torchcodec also under CUDA-enabled torch:
pip install --pre torchcodec --index-url https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/cpu
python -c "import torchcodec; print(torchcodec.__version__)"

# 2. Install torchaudio
printf "* Installing fsspec\n" # TODO: is this required for torchaudio??
pip install --trusted-host pypi.org --trusted-host files.pythonhosted.org fsspec
Expand Down
6 changes: 5 additions & 1 deletion .github/scripts/unittest-windows/run_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -12,5 +12,9 @@ python -m torch.utils.collect_env
env | grep TORCHAUDIO || true

cd test
pytest --continue-on-collection-errors --cov=torchaudio --junitxml=${RUNNER_TEST_RESULTS_DIR}/junit.xml -v --durations 20 torchaudio_unittest -k "not torchscript and not fairseq and not demucs and not librosa"
if [ -z "${CUDA_VERSION:-}" ] ; then
pytest --continue-on-collection-errors --cov=torchaudio --junitxml=${RUNNER_TEST_RESULTS_DIR}/junit.xml -v --durations 20 torchaudio_unittest -k "not torchscript and not fairseq and not demucs and not librosa"
else
pytest --continue-on-collection-errors --cov=torchaudio --junitxml=${RUNNER_TEST_RESULTS_DIR}/junit.xml -v --durations 20 torchaudio_unittest -k "not cpu and (cuda or gpu) and not torchscript and not fairseq and not demucs and not librosa"
fi
coverage html
2 changes: 1 addition & 1 deletion .github/scripts/unittest-windows/setup_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,4 +36,4 @@ fi
conda activate "${env_dir}"

# 3. Install minimal build tools
conda install -y -c conda-forge cmake ninja
conda install -y -c conda-forge cmake ninja "ffmpeg<8"
6 changes: 3 additions & 3 deletions .github/workflows/unittest-windows-cpu.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Unittests on Windows CPU
name: Unit-tests on Windows CPU

on:
pull_request:
Expand All @@ -10,7 +10,7 @@ on:
workflow_dispatch:

jobs:
unittests-windows-cpu:
build:
uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
with:
repository: pytorch/audio
Expand All @@ -21,7 +21,7 @@ jobs:
git config --global --add safe.directory /__w/audio/audio

# Set up Environment Variables
export PYTHON_VERSION="3.10"
export PYTHON_VERSION="3.11"
export PIP_PROGRESS_BAR=off
export CONDA_QUIET=1
unset CUDA_VERSION
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/unittest-windows-gpu.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Unittests on Windows GPU
name: Unit-tests on Windows GPU

on:
pull_request:
Expand All @@ -10,7 +10,7 @@ on:
workflow_dispatch:

jobs:
unittests-windows-gpu:
build:
uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
with:
repository: pytorch/audio
Expand Down
Loading