diff options
author | Adam J. Stewart <ajstewart426@gmail.com> | 2024-10-03 13:16:40 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-10-03 13:16:40 +0200 |
commit | feabcb884a87b7533318e38afe26df8c255f71b4 (patch) | |
tree | b636c4b82025da0098e272899165d8be3381cea6 | |
parent | 1c065f2da921193d3e8a0e1e9a49b57e9d866b73 (diff) | |
download | spack-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.py | 7 |
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 |