mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
Ensure version file is regenerated at change (#138237)
Fixes observed error where `version.py` would not be regenerated by CMake without deleting the file. Pull Request resolved: https://github.com/pytorch/pytorch/pull/138237 Approved by: https://github.com/Skylion007
This commit is contained in:
parent
770fcaf2ab
commit
2fc6c32b4c
|
|
@ -456,20 +456,15 @@ else()
|
|||
set(TORCH_VERSION_DEBUG 0)
|
||||
endif()
|
||||
|
||||
add_custom_command(
|
||||
OUTPUT ${TORCH_SRC_DIR}/version.py
|
||||
COMMAND "${CMAKE_COMMAND}" -E touch "${TOOLS_PATH}/generate_torch_version.py"
|
||||
COMMAND
|
||||
"${Python_EXECUTABLE}" "${TOOLS_PATH}/generate_torch_version.py"
|
||||
--is-debug=${TORCH_VERSION_DEBUG}
|
||||
--cuda-version=${CUDA_VERSION}
|
||||
--hip-version=${HIP_VERSION}
|
||||
DEPENDS ${TOOLS_PATH}/generate_torch_version.py
|
||||
WORKING_DIRECTORY ${TORCH_ROOT}
|
||||
)
|
||||
add_custom_target(
|
||||
gen_torch_version ALL
|
||||
DEPENDS ${TORCH_SRC_DIR}/version.py
|
||||
"${Python_EXECUTABLE}" "${TOOLS_PATH}/generate_torch_version.py"
|
||||
--is-debug=${TORCH_VERSION_DEBUG}
|
||||
--cuda-version=${CUDA_VERSION}
|
||||
--hip-version=${HIP_VERSION}
|
||||
BYPRODUCTS ${TORCH_SRC_DIR}/version.py
|
||||
COMMENT "Regenerating version file..."
|
||||
WORKING_DIRECTORY ${TORCH_ROOT}
|
||||
)
|
||||
add_dependencies(torch_python gen_torch_version)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user