pyrox.dev
1From 789ec77d09171f22d82006493cafbcf2496b4f8f Mon Sep 17 00:00:00 2001
2From: Someone Serge <sergei.kozlukov@aalto.fi>
3Date: Sat, 2 Dec 2023 12:47:00 +0000
4Subject: [PATCH] setup.py: propagate cmakeFlags
5
6---
7 tools/setup_helpers/extension.py | 3 +++
8 1 file changed, 3 insertions(+)
9
10diff --git a/tools/setup_helpers/extension.py b/tools/setup_helpers/extension.py
11index 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--
322.42.0