[Infra] Auto install pybind11-stubgen in cmake stage (#71911)

SigureMo · web-flow · commit ca53e2b81717 · 2025-03-26T23:15:22.000+08:00
diff --git a/paddle/fluid/pybind/CMakeLists.txt b/paddle/fluid/pybind/CMakeLists.txt
@@ -735,4 +735,23 @@ if(WITH_PYTHON)
   get_property(os_dependency_modules GLOBAL PROPERTY OS_DEPENDENCY_MODULES)
   target_link_libraries(${SHARD_LIB_NAME} ${os_dependency_modules})
 
+  function(install_py_pybind11_stubgen)
+    # check install
+    execute_process(
+      COMMAND
+        ${PYTHON_EXECUTABLE} "-c"
+        "import re, pybind11_stubgen; print(re.compile('/__init__.py.*').sub('',pybind11_stubgen.__file__))"
+      RESULT_VARIABLE _pybind11_stubgen_status
+      ERROR_QUIET OUTPUT_STRIP_TRAILING_WHITESPACE)
+
+    if(_pybind11_stubgen_status EQUAL 0)
+      return()
+    endif()
+
+    execute_process(COMMAND ${PYTHON_EXECUTABLE} -m pip install -U
+                            pybind11-stubgen)
+  endfunction()
+
+  install_py_pybind11_stubgen()
+
 endif()
diff --git a/python/requirements.txt b/python/requirements.txt
@@ -7,4 +7,3 @@ astor
 opt_einsum==3.3.0
 networkx
 typing_extensions
-pybind11-stubgen==2.5.1 # generate stub file from pybind11 C++ APIs [build]
