summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2024-10-03 13:16:40 +0200
committerGitHub <noreply@github.com>2024-10-03 13:16:40 +0200
commitfeabcb884a87b7533318e38afe26df8c255f71b4 (patch)
treeb636c4b82025da0098e272899165d8be3381cea6
parent1c065f2da921193d3e8a0e1e9a49b57e9d866b73 (diff)
downloadspack-feabcb884a87b7533318e38afe26df8c255f71b4.tar.gz
spack-feabcb884a87b7533318e38afe26df8c255f71b4.tar.bz2
spack-feabcb884a87b7533318e38afe26df8c255f71b4.tar.xz
spack-feabcb884a87b7533318e38afe26df8c255f71b4.zip
py-torchaudio: add support for CUDA 12.5+ (#46746)
-rw-r--r--var/spack/repos/builtin/packages/py-torchaudio/package.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/py-torchaudio/package.py b/var/spack/repos/builtin/packages/py-torchaudio/package.py
index e1e5236340..372cbca936 100644
--- a/var/spack/repos/builtin/packages/py-torchaudio/package.py
+++ b/var/spack/repos/builtin/packages/py-torchaudio/package.py
@@ -106,7 +106,12 @@ class PyTorchaudio(PythonPackage):
depends_on("sox")
# https://github.com/pytorch/audio/pull/3811
- depends_on("cuda@:12.4", when="^py-torch+cuda")
+ patch(
+ "https://github.com/pytorch/audio/pull/3811.patch?full_index=1",
+ sha256="34dce3403abb03f62827e8a1efcdb2bf7742477a01f155ebb9c7fefe9588b132",
+ when="@2.2:",
+ )
+ conflicts("^cuda@12.5:", when="@:2.1")
def setup_build_environment(self, env):
# tools/setup_helpers/extension.py