Fix cutlass build error when cuda greater than 11 and with_python=off (…

…PaddlePaddle#50763)
Xreki · Feb 23, 2023 · 22925c2 · 22925c2
1 parent 91a3d15
commit 22925c2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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}