frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

CMake build: preserve PYTHONPATH (#160144)

Browse Source 
Fixes #160092

I'm very new to CMake, so let me know if there's a fancier way to do this.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/160144
Approved by: https://github.com/malfet

Co-authored-by: Xuehai Pan <XuehaiPan@outlook.com>
This commit is contained in:
Adam J. Stewart
2025-08-08 16:03:49 +00:00
committed by PyTorch MergeBot
parent 50f23ff6f8
commit 231c72240d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/CMakeLists.txt
View File

@ -265,7 +265,7 @@ add_custom_command(
OUTPUT
"${TORCH_SRC_DIR}/utils/data/datapipes/datapipe.pyi"
COMMAND
${CMAKE_COMMAND} -E env PYTHONPATH="${TORCH_ROOT}"
${CMAKE_COMMAND} -E env --modify PYTHONPATH=path_list_prepend:"${TORCH_ROOT}" --
"${Python_EXECUTABLE}" ${TORCH_SRC_DIR}/utils/data/datapipes/gen_pyi.py
DEPENDS
"${TORCH_SRC_DIR}/utils/data/datapipes/datapipe.pyi.in"