Fix cutlass build error when cuda greater than 11 and with_python=off (#50763)

22925c21 · chalsliu · GitHub · 91a3d159 · 22925c21
显示空白变更内容
内联并排

Showing with 4 addition and 0 deletion

cmake/external/cutlass.cmake cmake/external/cutlass.cmake +4 -0

未找到文件。
--- a/cmake/external/cutlass.cmake
+++ b/cmake/external/cutlass.cmake
@@ -26,6 +26,10 @@ include_directories(

 add_definitions("-DPADDLE_WITH_CUTLASS")

+if(NOT PYTHON_EXECUTABLE)
+  find_package(PythonInterp REQUIRED)
+endif()
+
 ExternalProject_Add(
  extern_cutlass
  ${EXTERNAL_PROJECT_LOG_ARGS} ${SHALLOW_CLONE}