Skip to content

Commit 7de2d39

Browse files
authored
修改mlu编译脚本,适配多python版本编译 (#1372)
1 parent 51d41a2 commit 7de2d39

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

backends/mlu/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ add_custom_command(
121121

122122
add_custom_command(
123123
OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/python/.timestamp
124-
COMMAND python3 ${CMAKE_CURRENT_BINARY_DIR}/setup.py bdist_wheel
124+
COMMAND ${Python_EXECUTABLE} ${CMAKE_CURRENT_BINARY_DIR}/setup.py bdist_wheel
125125
DEPENDS ${CUSTOM_MLU_NAME}
126126
COMMENT "Packing whl packages------>>>")
127127

backends/mlu/tools/compile.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ Configuring cmake in build ...
4040
-DWITH_MKL=${WITH_MKL}
4141
-DWITH_ARM=${WITH_ARM}
4242
-DON_INFER=${ON_INFER:-OFF}
43+
-DPYTHON_VERSION=${PYTHON_VERSION:-""}
4344
========================================
4445
EOF
4546

@@ -51,6 +52,7 @@ cmake .. \
5152
-DWITH_MKL=${WITH_MKL:-ON} \
5253
-DWITH_ARM=${WITH_ARM:-OFF} \
5354
-DON_INFER=${ON_INFER:-OFF} \
55+
-DPYTHON_VERSION=${PYTHON_VERSION:-""} \
5456
-DCMAKE_EXPORT_COMPILE_COMMANDS=ON;cmake_error=$?
5557

5658
if [ "$cmake_error" != 0 ];then

0 commit comments

Comments
 (0)