支持多python版本编译 (#1371)

Author: XieYunshen

backends/npu/CMakeLists.txt

@@ -158,7 +158,7 @@ add_custom_command(
           ${CMAKE_CURRENT_BINARY_DIR}/python/paddle_custom_device/npu/passes
   COMMAND ${CMAKE_COMMAND} -E copy_if_different ${CMAKE_SOURCE_DIR}/passes/*
           ${CMAKE_CURRENT_BINARY_DIR}/python/paddle_custom_device/npu/passes
-  COMMAND python3 ${CMAKE_CURRENT_BINARY_DIR}/setup.py bdist_wheel
+  COMMAND ${Python_EXECUTABLE} ${CMAKE_CURRENT_BINARY_DIR}/setup.py bdist_wheel
   DEPENDS ${CUSTOM_NPU_NAME}
   COMMENT "Packing whl packages------>>>")
 

backends/npu/tools/compile.sh

@@ -46,6 +46,7 @@ Configuring cmake in build ...
     -DWITH_ATB=${WITH_ATB:-OFF}
     -DON_INFER=${ON_INFER:-OFF}
     -DWITH_COVERAGE=${WITH_COVERAGE:-OFF}
+    -DPYTHON_VERSION=${PYTHON_VERSION:-""}
 ========================================
 EOF
 
@@ -58,6 +59,7 @@ cmake .. \
     -DWITH_ATB=${WITH_ATB:-OFF} \
     -DON_INFER=${ON_INFER:-OFF} \
     -DWITH_COVERAGE=${WITH_COVERAGE:-OFF} \
+    -DPYTHON_VERSION=${PYTHON_VERSION:-""} \
     -DCMAKE_EXPORT_COMPILE_COMMANDS=ON;cmake_error=$?
 
 if [ "$cmake_error" != 0 ];then

cmake/paddle.cmake

@@ -12,8 +12,16 @@
 # License for the specific language governing permissions and limitations under
 # the License.
 
-find_package(Python ${PYTHON_VERSION} REQUIRED COMPONENTS Interpreter
-                                                          Development)
+if(NOT PYTHON_VERSION)
+  find_package(Python REQUIRED COMPONENTS Interpreter Development)
+else()
+  find_package(
+    Python ${PYTHON_VERSION} REQUIRED
+    COMPONENTS Interpreter Development
+    EXACT)
+endif()
+
+message(STATUS "Python_EXECUTABLE is ${Python_EXECUTABLE}")
 include_directories(${Python_INCLUDE_DIRS})
 
 if(DEFINED ENV{PADDLE_CUSTOM_PATH})