forked from mindspore-Ecosystem/mindspore
91 lines
2.8 KiB
CMake
91 lines
2.8 KiB
CMake
# find exec
|
|
find_package(Python3 3.7 COMPONENTS Interpreter Development)
|
|
if (NOT Python3_FOUND)
|
|
message("No python3 found.")
|
|
return ()
|
|
endif ()
|
|
|
|
set(PYTHON ${Python3_EXECUTABLE})
|
|
set(PYTHON_VERSION ${Python3_VERSION_MAJOR}.${Python3_VERSION_MINOR})
|
|
|
|
find_package(Git)
|
|
if (NOT GIT_FOUND)
|
|
message("No git found.")
|
|
return ()
|
|
endif ()
|
|
set(GIT ${GIT_EXECUTABLE})
|
|
|
|
# set path
|
|
set(MS_ROOT_DIR ${CPACK_PACKAGE_DIRECTORY}/../../)
|
|
set(MS_PACK_ROOT_DIR ${MS_ROOT_DIR}/build/package)
|
|
|
|
# set package file name
|
|
if (CMAKE_SYSTEM_NAME MATCHES "Linux")
|
|
if (PYTHON_VERSION MATCHES "3.7")
|
|
set(PY_TAGS "cp37-cp37m")
|
|
else ()
|
|
message("Could not find 'Python 3.7'")
|
|
return()
|
|
endif ()
|
|
string(TOLOWER linux_${CMAKE_HOST_SYSTEM_PROCESSOR} PLATFORM_TAG)
|
|
elseif (CMAKE_SYSTEM_NAME MATCHES "Darwin")
|
|
if (PYTHON_VERSION MATCHES "3.7")
|
|
set(PY_TAGS "py37-none")
|
|
else ()
|
|
message("Could not find 'Python 3.7'")
|
|
return()
|
|
endif ()
|
|
set(PLATFORM_TAG "any")
|
|
elseif (CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|
if (PYTHON_VERSION MATCHES "3.7")
|
|
set(PY_TAGS "cp37-cp37m")
|
|
else ()
|
|
message("Could not find 'Python 3.7'")
|
|
return()
|
|
endif ()
|
|
set(PLATFORM_TAG "win_amd64")
|
|
else ()
|
|
message(FATAL_ERROR "other platform: ${CMAKE_SYSTEM_NAME}")
|
|
endif ()
|
|
|
|
# get git commit id
|
|
set(GIT_COMMIT_ID "")
|
|
execute_process(
|
|
COMMAND ${GIT} log --format='[sha1]:%h,[branch]:%d' -1
|
|
OUTPUT_VARIABLE GIT_COMMIT_ID
|
|
WORKING_DIRECTORY ${MS_ROOT_DIR}
|
|
ERROR_QUIET)
|
|
string(REPLACE " " "" GIT_COMMIT_ID ${GIT_COMMIT_ID})
|
|
|
|
set(ENV{BACKEND_POLICY} ${CPACK_MS_BACKEND})
|
|
set(ENV{MS_PACKAGE_NAME} ${CPACK_MS_PACKAGE_NAME})
|
|
set(ENV{COMMIT_ID} ${GIT_COMMIT_ID})
|
|
|
|
execute_process(
|
|
COMMAND ${PYTHON} ${MS_ROOT_DIR}/setup.py "bdist_wheel"
|
|
WORKING_DIRECTORY ${MS_PACK_ROOT_DIR}
|
|
)
|
|
|
|
# finally
|
|
set(PACKAGE_NAME ${CPACK_MS_PACKAGE_NAME})
|
|
if (NOT CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|
string(REPLACE "-" "_" PACKAGE_NAME ${PACKAGE_NAME})
|
|
execute_process(
|
|
COMMAND chmod -R 700 ${MS_PACK_ROOT_DIR}/mindspore/
|
|
COMMAND chmod -R 700 ${MS_PACK_ROOT_DIR}/${PACKAGE_NAME}.egg-info/
|
|
)
|
|
endif ()
|
|
|
|
file(GLOB WHL_FILE ${MS_PACK_ROOT_DIR}/dist/*.whl)
|
|
get_filename_component(ORIGIN_FILE_NAME ${WHL_FILE} NAME)
|
|
string(REPLACE "-" ";" ORIGIN_FILE_NAME ${ORIGIN_FILE_NAME})
|
|
list(GET ORIGIN_FILE_NAME 1 VERSION)
|
|
set(NEW_FILE_NAME ${PACKAGE_NAME}-${VERSION}-${PY_TAGS}-${PLATFORM_TAG}.whl)
|
|
file(RENAME ${WHL_FILE} ${MS_PACK_ROOT_DIR}/${NEW_FILE_NAME})
|
|
file(REMOVE_RECURSE ${MS_ROOT_DIR}/output)
|
|
file(MAKE_DIRECTORY ${MS_ROOT_DIR}/output)
|
|
file(COPY ${MS_PACK_ROOT_DIR}/${NEW_FILE_NAME} DESTINATION ${MS_ROOT_DIR}/output/)
|
|
|
|
file(SHA256 ${MS_ROOT_DIR}/output/${NEW_FILE_NAME} SHA256_VAR)
|
|
file(WRITE ${MS_ROOT_DIR}/output/${NEW_FILE_NAME}.sha256 ${SHA256_VAR} " " ${NEW_FILE_NAME})
|