pyrox.dev
1From 10b7e8330bdba319a4162cceb8e5dd4280215b04 Mon Sep 17 00:00:00 2001
2From: SomeoneSerge <else@someonex.net>
3Date: Wed, 31 Jul 2024 12:06:15 +0000
4Subject: [PATCH 2/2] setup.py: nix-support (respect cmakeFlags)
5
6---
7 setup.py | 10 ++++++++++
8 1 file changed, 10 insertions(+)
9
10diff --git a/setup.py b/setup.py
11index 01e006f9..14762146 100644
12--- a/setup.py
13+++ b/setup.py
14@@ -15,6 +15,15 @@ from setuptools import Extension, find_packages, setup
15 from setuptools.command.build_ext import build_ext
16 from torch.utils.cpp_extension import CUDA_HOME
17
18+import os
19+import json
20+
21+if "NIX_ATTRS_JSON_FILE" in os.environ:
22+ with open(os.environ["NIX_ATTRS_JSON_FILE"], "r") as f:
23+ NIX_ATTRS = json.load(f)
24+else:
25+ NIX_ATTRS = { "cmakeFlags": os.environ.get("cmakeFlags", "").split() }
26+
27
28 def load_module_from_path(module_name, path):
29 spec = importlib.util.spec_from_file_location(module_name, path)
30@@ -159,6 +168,7 @@ class cmake_build_ext(build_ext):
31 '-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={}'.format(outdir),
32 '-DCMAKE_ARCHIVE_OUTPUT_DIRECTORY={}'.format(self.build_temp),
33 '-DVLLM_TARGET_DEVICE={}'.format(VLLM_TARGET_DEVICE),
34+ *NIX_ATTRS["cmakeFlags"],
35 ]
36
37 verbose = envs.VERBOSE
38--
392.45.1
40