mirror of
https://github.com/NixOS/nixpkgs.git
synced 2025-02-23 04:25:14 +00:00
1b6efd0fca
Diff: https://github.com/pytorch/pytorch/compare/v2.3.1...v2.4.0 Changelog: https://github.com/pytorch/pytorch/releases/tag/v2.4.0 Co-authored-by: =?UTF-8?q?Danie=CC=88l=20de=20Kok?= <me@danieldk.eu> Co-authored-by: Connor Baker <connor.baker@tweag.io>
13 lines
430 B
Diff
13 lines
430 B
Diff
diff --git a/tools/setup_helpers/cmake.py b/tools/setup_helpers/cmake.py
|
|
index 5481ce46031c..d50d9d547399 100644
|
|
--- a/tools/setup_helpers/cmake.py
|
|
+++ b/tools/setup_helpers/cmake.py
|
|
@@ -231,6 +231,7 @@ def generate(
|
|
"SELECTED_OP_LIST",
|
|
"TORCH_CUDA_ARCH_LIST",
|
|
"TRACING_BASED",
|
|
+ "PYTHON_LIB_REL_PATH",
|
|
)
|
|
}
|
|
)
|