pyrox.dev
Merge pull request #272052 from GaetanLepage/torchaudio
python311Packages.torchaudio: fix build when cudaSupport is enabled
authored by Martin Weinelt and committed by GitHub 739d72c5 ba513b29
+32
pkgs/development/python-modules/torchaudio/0001-setup.py-propagate-cmakeFlags.patch
+32
pkgs/development/python-modules/torchaudio/0001-setup.py-propagate-cmakeFlags.patch
···
···
1
+
From 789ec77d09171f22d82006493cafbcf2496b4f8f Mon Sep 17 00:00:00 2001
2
+
From: Someone Serge <sergei.kozlukov@aalto.fi>
3
+
Date: Sat, 2 Dec 2023 12:47:00 +0000
4
+
Subject: [PATCH] setup.py: propagate cmakeFlags
5
+
6
+
---
7
+
tools/setup_helpers/extension.py | 3 +++
8
+
1 file changed, 3 insertions(+)
9
+
10
+
diff --git a/tools/setup_helpers/extension.py b/tools/setup_helpers/extension.py
11
+
index 2415bbae..7c2a9351 100644
12
+
--- a/tools/setup_helpers/extension.py
13
+
+++ b/tools/setup_helpers/extension.py
14
+
@@ -1,6 +1,7 @@
15
+
import distutils.sysconfig
16
+
import os
17
+
import platform
18
+
+import shlex
19
+
import subprocess
20
+
from pathlib import Path
21
+
22
+
@@ -141,6 +142,8 @@ class CMakeBuild(build_ext):
23
+
f"-DUSE_OPENMP:BOOL={'ON' if _USE_OPENMP else 'OFF'}",
24
+
f"-DUSE_FFMPEG:BOOL={'ON' if _USE_FFMPEG else 'OFF'}",
25
+
]
26
+
+ if "cmakeFlags" in os.environ:
27
+
+ cmake_args.extend(shlex.split(os.environ["cmakeFlags"]))
28
+
build_args = ["--target", "install"]
29
+
# Pass CUDA architecture to cmake
30
+
if _TORCH_CUDA_ARCH_LIST is not None:
31
+
--
32
+
2.42.0
+4
pkgs/development/python-modules/torchaudio/default.nix
+4
pkgs/development/python-modules/torchaudio/default.nix