Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
76 commits
Select commit Hold shift + click to select a range
2e25279
Add torchcodec mock with wav loading and saving
samanklesaria Jul 18, 2025
fe375f4
Merge branch 'main' into test_wav_hack
NicolasHug Jul 28, 2025
dd90ff3
WIP
samanklesaria Aug 11, 2025
d2ccd82
Remove libtorio ffmpeg from cmakelists
samanklesaria Aug 11, 2025
7b47628
Remove io directory
samanklesaria Aug 11, 2025
a300221
Let load and save rely on *_with_torchcodec
NicolasHug Jul 16, 2025
07e3b77
install torchcodec in doc job
NicolasHug Jul 16, 2025
92719d3
Add docstring and arguments for load and save
samanklesaria Aug 12, 2025
4a98ee5
Revise docstring
samanklesaria Aug 13, 2025
7b02754
Add typing imports
samanklesaria Aug 13, 2025
74edc0a
Try ffmpeg>4
samanklesaria Aug 13, 2025
80f5eb7
Install conda deps before pip deps
samanklesaria Aug 13, 2025
7f063a6
Add scipy hack for load and save
samanklesaria Aug 13, 2025
700c6c9
Only import scipy during testing
samanklesaria Aug 13, 2025
6995b21
Revert "Install conda deps before pip deps"
samanklesaria Aug 13, 2025
4ab5993
Revert "Try ffmpeg>4"
samanklesaria Aug 13, 2025
43c4602
Revert torchcodec installation changes
samanklesaria Aug 13, 2025
f74f004
Use existing wav_utils
samanklesaria Aug 13, 2025
89ca133
Remove _backend folder
samanklesaria Aug 13, 2025
50a538a
Merge branch 'remove_backends' into remove_io_dir
samanklesaria Aug 13, 2025
3bd3038
Merge branch 'remove_io_dir' into remove_libtorio
samanklesaria Aug 13, 2025
953fc65
Support frame_offset and num_frames in load hack
samanklesaria Aug 13, 2025
9ab06f7
Merge branch 'torchcodec_loading' into remove_backends
samanklesaria Aug 13, 2025
bea8d82
Merge branch 'remove_backends' into remove_io_dir
samanklesaria Aug 13, 2025
dd3ff90
Use rand instead of randn for test_save_channels_first
samanklesaria Aug 14, 2025
6013c36
Merge branch 'torchcodec_loading' into remove_backends
samanklesaria Aug 14, 2025
872310b
Merge branch 'remove_backends' into remove_io_dir
samanklesaria Aug 14, 2025
6e3f83e
Merge branch 'remove_io_dir' into remove_libtorio
samanklesaria Aug 14, 2025
72539b9
Merge branch 'test_wav_hack' into torchcodec_loading
samanklesaria Aug 14, 2025
c94e011
Remove pytest-aware code in src
samanklesaria Aug 14, 2025
b622d82
Remove torchcodec version check
samanklesaria Aug 14, 2025
93351a2
Fix bugs in torchcodec mock
samanklesaria Aug 14, 2025
5407163
Skip test_load_save_torchcodec
samanklesaria Aug 14, 2025
bd7eb52
Correct call to pytest skip
samanklesaria Aug 14, 2025
c3d0cc2
Remove torchcodec installation
samanklesaria Aug 14, 2025
d10fc19
Add torchcodec to build installation
samanklesaria Aug 15, 2025
92fee51
Remove redundant wav_utils
samanklesaria Aug 15, 2025
34d39ed
Merge branch 'torchcodec_loading' into remove_backends
samanklesaria Aug 15, 2025
0f6ce2d
Merge branch 'remove_backends' into remove_io_dir
samanklesaria Aug 15, 2025
a178f93
Merge branch 'remove_io_dir' into remove_libtorio
samanklesaria Aug 15, 2025
03b441e
Merge branch 'main' into remove_backends
samanklesaria Aug 18, 2025
4e0a9f1
Merge branch 'remove_backends' into remove_io_dir
samanklesaria Aug 18, 2025
ec4712d
Merge branch 'remove_io_dir' into remove_libtorio
samanklesaria Aug 18, 2025
8ac0720
Remove io export
samanklesaria Aug 18, 2025
48c7e95
Merge branch 'remove_io_dir' into remove_libtorio
samanklesaria Aug 18, 2025
6b7d78c
Remove io import
samanklesaria Aug 18, 2025
f2c21e6
Remove torchaudio.io references in docs
samanklesaria Aug 18, 2025
3e6cd31
Merge branch 'remove_io_dir' into remove_libtorio
samanklesaria Aug 18, 2025
166c123
Remove prototype files
samanklesaria Aug 19, 2025
530c036
Remove deprecation test for prototype feature
samanklesaria Aug 19, 2025
c915818
Remove ffmpeg_utils
samanklesaria Aug 19, 2025
4070a8d
Remove additional references to ffmpeg_util
samanklesaria Aug 19, 2025
750d39e
Merge branch 'main' into remove_prototype
samanklesaria Aug 19, 2025
3794da6
Remove references to torchaudio.prototype
samanklesaria Aug 19, 2025
72fcf79
Merge branch 'main' into remove_ffmpeg_util
samanklesaria Aug 19, 2025
11560ef
Merge branch 'main' into remove_libtorio
samanklesaria Aug 19, 2025
7ef8c69
Remove some torio references
samanklesaria Aug 19, 2025
b184873
Merge branch 'main' into remove_libtorio
samanklesaria Aug 19, 2025
a9123a9
Remove docs references to libtorio
samanklesaria Aug 19, 2025
6e1ff7e
FFmpeg available missing
samanklesaria Aug 19, 2025
075b613
Merge branch 'remove_ffmpeg_util' into remove_libtorio
samanklesaria Aug 19, 2025
4257c04
Merge branch 'main' into remove_ffmpeg_util
samanklesaria Aug 19, 2025
c001763
Remove _IS_FFMPEG_AVAILABLE references
samanklesaria Aug 19, 2025
4533e1f
Merge branch 'remove_ffmpeg_util' into remove_libtorio
samanklesaria Aug 19, 2025
4035cd1
Remove kaldi_io module
samanklesaria Aug 20, 2025
b0650da
Remove deprecation of cutc decoder
samanklesaria Aug 20, 2025
954bb38
Merge branch 'remove_prototype' into remove_dropping_support
samanklesaria Aug 20, 2025
96ef204
Merge branch 'remove_kaldi_io_mod' into remove_dropping_support
samanklesaria Aug 20, 2025
126a2b0
Merge branch 'no_wrap_ctc_decoder' into remove_dropping_support
samanklesaria Aug 20, 2025
9a50365
Remove dropping_support
samanklesaria Aug 20, 2025
0607381
Enable windows tests
samanklesaria Aug 20, 2025
8edabb3
Simplify windows install script
samanklesaria Aug 20, 2025
2e2421d
Remove last dropping_support call
samanklesaria Aug 20, 2025
22965e4
Merge branch 'remove_dropping_support' into enable_windows_tests
samanklesaria Aug 20, 2025
c1e3840
Import rnnt correctly
samanklesaria Aug 20, 2025
9db1db1
Merge branch 'remove_dropping_support' into enable_windows_tests
samanklesaria Aug 20, 2025
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
1 change: 0 additions & 1 deletion .github/scripts/unittest-windows/environment.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@ dependencies:
- scipy >= 1.4.1
- pip
- pip:
- PySoundFile
- future
- parameterized
- dataclasses
Expand Down
43 changes: 1 addition & 42 deletions .github/scripts/unittest-windows/install.sh
Original file line number Diff line number Diff line change
Expand Up @@ -50,45 +50,4 @@ printf "* Installing torchaudio\n"

# 3. Install Test tools
printf "* Installing test tools\n"
NUMBA_DEV_CHANNEL=""
SENTENCEPIECE_DEPENDENCY="sentencepiece"
case "$(python --version)" in
*3.9*)
# Numba isn't available for Python 3.9 except on the numba dev channel and building from source fails
# See https://github.com/librosa/librosa/issues/1270#issuecomment-759065048
NUMBA_DEV_CHANNEL="-c numba/label/dev"
;;
*3.10*)
# Don't install sentencepiece, no python 3.10 dependencies available for windows yet
SENTENCEPIECE_DEPENDENCY=""
NUMBA_DEV_CHANNEL="-c numba/label/dev"
;;
esac
(
conda install -y -c conda-forge ${NUMBA_DEV_CHANNEL} parameterized 'requests>=2.20'
# Need to disable shell check since this'll fail out if SENTENCEPIECE_DEPENDENCY is empty
# shellcheck disable=SC2086
pip install \
${SENTENCEPIECE_DEPENDENCY} \
Pillow \
SoundFile \
coverage \
expecttest \
inflect \
pytest \
pytest-cov \
pytorch-lightning \
'scipy==1.7.3' \
unidecode \
'protobuf<4.21.0' \
demucs \
tinytag \
pyroomacoustics \
flashlight-text \
git+https://github.com/kpu/kenlm/
)
# Install fairseq
git clone https://github.com/pytorch/fairseq
cd fairseq
git checkout e47a4c8
pip install .
pip install parameterized requests coverage pytest pytest-cov scipy numpy expecttest
92 changes: 46 additions & 46 deletions .github/workflows/unittest-windows-cpu.yml
Original file line number Diff line number Diff line change
@@ -1,52 +1,52 @@
# name: Unittests on Windows CPU
name: Unittests on Windows CPU

# on:
# pull_request:
# push:
# branches:
# - nightly
# - main
# - release/*
# workflow_dispatch:
on:
pull_request:
push:
branches:
- nightly
- main
- release/*
workflow_dispatch:

# jobs:
# unittests-windows-cpu:
# uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
# with:
# repository: pytorch/audio
# runner: windows.4xlarge
# timeout: 180
# script: |
# # Mark Build Directory Safe
# git config --global --add safe.directory /__w/audio/audio
jobs:
unittests-windows-cpu:
uses: pytorch/test-infra/.github/workflows/windows_job.yml@main
with:
repository: pytorch/audio
runner: windows.4xlarge
timeout: 180
script: |
# Mark Build Directory Safe
git config --global --add safe.directory /__w/audio/audio

# # Set up Environment Variables
# export PYTHON_VERSION="3.9"
# unset CUDA_VERSION
# Set up Environment Variables
export PYTHON_VERSION="3.9"
unset CUDA_VERSION

# # 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
# 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_CUDA=true
# export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_HW_ACCEL=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_TEMPORARY_DISABLED=true
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_CUDA=true
export TORCHAUDIO_TEST_ALLOW_SKIP_IF_NO_HW_ACCEL=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_TEMPORARY_DISABLED=true

# .github/scripts/unittest-windows/setup_env.sh
# .github/scripts/unittest-windows/install.sh
# .github/scripts/unittest-windows/run_test.sh
.github/scripts/unittest-windows/setup_env.sh
.github/scripts/unittest-windows/install.sh
.github/scripts/unittest-windows/run_test.sh
90 changes: 45 additions & 45 deletions .github/workflows/unittest-windows-gpu.yml
Original file line number Diff line number Diff line change
@@ -1,52 +1,52 @@
# name: Unittests on Windows GPU
name: Unittests on Windows GPU

# on:
# pull_request:
# push:
# branches:
# - nightly
# - main
# - release/*
# workflow_dispatch:
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
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.9"
# export CUDA_VERSION=12.6
# Set up Environment Variables
export PYTHON_VERSION="3.9"
export CUDA_VERSION=12.6

# # 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
# 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
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
# .github/scripts/unittest-windows/install.sh
# .github/scripts/unittest-windows/run_test.sh
.github/scripts/unittest-windows/setup_env.sh
.github/scripts/unittest-windows/install.sh
.github/scripts/unittest-windows/run_test.sh
1 change: 0 additions & 1 deletion CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,6 @@ if (USE_FFMPEG)
message(STATUS "Building FFmpeg integration with multi version support")
add_subdirectory(third_party/ffmpeg/multi)
endif()
add_subdirectory(src/libtorio/ffmpeg)
endif()
if (BUILD_CUDA_CTC_DECODER)
if (NOT USE_CUDA)
Expand Down
1 change: 0 additions & 1 deletion CONTRIBUTING.md
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,6 @@ Optional packages to install if you want to run related tests:
- `librosa`
- `requests`
- `soundfile`
- `kaldi_io`
- `transformers`
- `fairseq` (it has to be newer than `0.10.2`, so you will need to install from
source. Commit `e6eddd80` is known to work.)
Expand Down
21 changes: 0 additions & 21 deletions cmake/TorchAudioHelper.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -41,17 +41,6 @@ function(torchaudio_library name source include_dirs link_libraries compile_defs
)
endfunction()

function(torio_library name source include_dirs link_libraries compile_defs)
_library(
torio/lib
"${name}"
"${source}"
"${include_dirs}"
"${link_libraries}"
"${compile_defs}"
)
endfunction()

if (BUILD_TORCHAUDIO_PYTHON_EXTENSION)
# See https://github.com/pytorch/pytorch/issues/38122
find_library(TORCH_PYTHON_LIBRARY torch_python PATHS "${TORCH_INSTALL_PREFIX}/lib")
Expand Down Expand Up @@ -103,16 +92,6 @@ if (BUILD_TORCHAUDIO_PYTHON_EXTENSION)
"${definitions}"
)
endfunction()
function(torio_extension name sources include_dirs libraries definitions)
_extension(
torio/lib
"${name}"
"${sources}"
"${include_dirs}"
"${libraries}"
"${definitions}"
)
endfunction()
endif()


Expand Down
1 change: 0 additions & 1 deletion docs/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,6 @@ docset: html
# Catch-all target: route all unknown targets to Sphinx using the new
# "make mode" option. $(O) is meant as a shortcut for $(SPHINXOPTS).
%: Makefile
# doxygen source/Doxyfile
@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
@python post_process_dispatcher.py $(BUILDDIR)

Expand Down
Loading
Loading