159e378cbb
GitOrigin-RevId: c04d5652cfa9742b1d519688f65d1bbccea9eb7e
12 lines
430 B
Diff
12 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",
|
|
)
|
|
}
|
|
)
|