The source code and dockerfile for the GSW2024 AI Lab.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.

194 lines
7.8 KiB

4 weeks ago
  1. # - Find python libraries
  2. # This module finds the libraries corresponding to the Python interpeter
  3. # FindPythonInterp provides.
  4. # This code sets the following variables:
  5. #
  6. # PYTHONLIBS_FOUND - have the Python libs been found
  7. # PYTHON_PREFIX - path to the Python installation
  8. # PYTHON_LIBRARIES - path to the python library
  9. # PYTHON_INCLUDE_DIRS - path to where Python.h is found
  10. # PYTHON_MODULE_EXTENSION - lib extension, e.g. '.so' or '.pyd'
  11. # PYTHON_MODULE_PREFIX - lib name prefix: usually an empty string
  12. # PYTHON_SITE_PACKAGES - path to installation site-packages
  13. # PYTHON_IS_DEBUG - whether the Python interpreter is a debug build
  14. #
  15. # Thanks to talljimbo for the patch adding the 'LDVERSION' config
  16. # variable usage.
  17. #=============================================================================
  18. # Copyright 2001-2009 Kitware, Inc.
  19. # Copyright 2012 Continuum Analytics, Inc.
  20. #
  21. # All rights reserved.
  22. #
  23. # Redistribution and use in source and binary forms, with or without
  24. # modification, are permitted provided that the following conditions
  25. # are met:
  26. #
  27. # * Redistributions of source code must retain the above copyright
  28. # notice, this list of conditions and the following disclaimer.
  29. #
  30. # * Redistributions in binary form must reproduce the above copyright
  31. # notice, this list of conditions and the following disclaimer in the
  32. # documentation and/or other materials provided with the distribution.
  33. #
  34. # * Neither the names of Kitware, Inc., the Insight Software Consortium,
  35. # nor the names of their contributors may be used to endorse or promote
  36. # products derived from this software without specific prior written
  37. # permission.
  38. #
  39. # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  40. # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  41. # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  42. # # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  43. # HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  44. # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  45. # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  46. # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  47. # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  48. # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  49. # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  50. #=============================================================================
  51. if(PYTHONLIBS_FOUND)
  52. return()
  53. endif()
  54. # Use the Python interpreter to find the libs.
  55. if(PythonLibsNew_FIND_REQUIRED)
  56. find_package(PythonInterp ${PythonLibsNew_FIND_VERSION} REQUIRED)
  57. else()
  58. find_package(PythonInterp ${PythonLibsNew_FIND_VERSION})
  59. endif()
  60. if(NOT PYTHONINTERP_FOUND)
  61. set(PYTHONLIBS_FOUND FALSE)
  62. return()
  63. endif()
  64. # According to http://stackoverflow.com/questions/646518/python-how-to-detect-debug-interpreter
  65. # testing whether sys has the gettotalrefcount function is a reliable, cross-platform
  66. # way to detect a CPython debug interpreter.
  67. #
  68. # The library suffix is from the config var LDVERSION sometimes, otherwise
  69. # VERSION. VERSION will typically be like "2.7" on unix, and "27" on windows.
  70. execute_process(COMMAND "${PYTHON_EXECUTABLE}" "-c"
  71. "from distutils import sysconfig as s;import sys;import struct;
  72. print('.'.join(str(v) for v in sys.version_info));
  73. print(sys.prefix);
  74. print(s.get_python_inc(plat_specific=True));
  75. print(s.get_python_lib(plat_specific=True));
  76. print(s.get_config_var('SO'));
  77. print(hasattr(sys, 'gettotalrefcount')+0);
  78. print(struct.calcsize('@P'));
  79. print(s.get_config_var('LDVERSION') or s.get_config_var('VERSION'));
  80. print(s.get_config_var('LIBDIR') or '');
  81. print(s.get_config_var('MULTIARCH') or '');
  82. "
  83. RESULT_VARIABLE _PYTHON_SUCCESS
  84. OUTPUT_VARIABLE _PYTHON_VALUES
  85. ERROR_VARIABLE _PYTHON_ERROR_VALUE)
  86. if(NOT _PYTHON_SUCCESS MATCHES 0)
  87. if(PythonLibsNew_FIND_REQUIRED)
  88. message(FATAL_ERROR
  89. "Python config failure:\n${_PYTHON_ERROR_VALUE}")
  90. endif()
  91. set(PYTHONLIBS_FOUND FALSE)
  92. return()
  93. endif()
  94. # Convert the process output into a list
  95. string(REGEX REPLACE ";" "\\\\;" _PYTHON_VALUES ${_PYTHON_VALUES})
  96. string(REGEX REPLACE "\n" ";" _PYTHON_VALUES ${_PYTHON_VALUES})
  97. list(GET _PYTHON_VALUES 0 _PYTHON_VERSION_LIST)
  98. list(GET _PYTHON_VALUES 1 PYTHON_PREFIX)
  99. list(GET _PYTHON_VALUES 2 PYTHON_INCLUDE_DIR)
  100. list(GET _PYTHON_VALUES 3 PYTHON_SITE_PACKAGES)
  101. list(GET _PYTHON_VALUES 4 PYTHON_MODULE_EXTENSION)
  102. list(GET _PYTHON_VALUES 5 PYTHON_IS_DEBUG)
  103. list(GET _PYTHON_VALUES 6 PYTHON_SIZEOF_VOID_P)
  104. list(GET _PYTHON_VALUES 7 PYTHON_LIBRARY_SUFFIX)
  105. list(GET _PYTHON_VALUES 8 PYTHON_LIBDIR)
  106. list(GET _PYTHON_VALUES 9 PYTHON_MULTIARCH)
  107. # Make sure the Python has the same pointer-size as the chosen compiler
  108. # Skip if CMAKE_SIZEOF_VOID_P is not defined
  109. if(CMAKE_SIZEOF_VOID_P AND (NOT "${PYTHON_SIZEOF_VOID_P}" STREQUAL "${CMAKE_SIZEOF_VOID_P}"))
  110. if(PythonLibsNew_FIND_REQUIRED)
  111. math(EXPR _PYTHON_BITS "${PYTHON_SIZEOF_VOID_P} * 8")
  112. math(EXPR _CMAKE_BITS "${CMAKE_SIZEOF_VOID_P} * 8")
  113. message(FATAL_ERROR
  114. "Python config failure: Python is ${_PYTHON_BITS}-bit, "
  115. "chosen compiler is ${_CMAKE_BITS}-bit")
  116. endif()
  117. set(PYTHONLIBS_FOUND FALSE)
  118. return()
  119. endif()
  120. # The built-in FindPython didn't always give the version numbers
  121. string(REGEX REPLACE "\\." ";" _PYTHON_VERSION_LIST ${_PYTHON_VERSION_LIST})
  122. list(GET _PYTHON_VERSION_LIST 0 PYTHON_VERSION_MAJOR)
  123. list(GET _PYTHON_VERSION_LIST 1 PYTHON_VERSION_MINOR)
  124. list(GET _PYTHON_VERSION_LIST 2 PYTHON_VERSION_PATCH)
  125. # Make sure all directory separators are '/'
  126. string(REGEX REPLACE "\\\\" "/" PYTHON_PREFIX ${PYTHON_PREFIX})
  127. string(REGEX REPLACE "\\\\" "/" PYTHON_INCLUDE_DIR ${PYTHON_INCLUDE_DIR})
  128. string(REGEX REPLACE "\\\\" "/" PYTHON_SITE_PACKAGES ${PYTHON_SITE_PACKAGES})
  129. if(CMAKE_HOST_WIN32)
  130. set(PYTHON_LIBRARY
  131. "${PYTHON_PREFIX}/libs/Python${PYTHON_LIBRARY_SUFFIX}.lib")
  132. # when run in a venv, PYTHON_PREFIX points to it. But the libraries remain in the
  133. # original python installation. They may be found relative to PYTHON_INCLUDE_DIR.
  134. if(NOT EXISTS "${PYTHON_LIBRARY}")
  135. get_filename_component(_PYTHON_ROOT ${PYTHON_INCLUDE_DIR} DIRECTORY)
  136. set(PYTHON_LIBRARY
  137. "${_PYTHON_ROOT}/libs/Python${PYTHON_LIBRARY_SUFFIX}.lib")
  138. endif()
  139. # raise an error if the python libs are still not found.
  140. if(NOT EXISTS "${PYTHON_LIBRARY}")
  141. message(FATAL_ERROR "Python libraries not found")
  142. endif()
  143. else()
  144. if(PYTHON_MULTIARCH)
  145. set(_PYTHON_LIBS_SEARCH "${PYTHON_LIBDIR}/${PYTHON_MULTIARCH}" "${PYTHON_LIBDIR}")
  146. else()
  147. set(_PYTHON_LIBS_SEARCH "${PYTHON_LIBDIR}")
  148. endif()
  149. #message(STATUS "Searching for Python libs in ${_PYTHON_LIBS_SEARCH}")
  150. # Probably this needs to be more involved. It would be nice if the config
  151. # information the python interpreter itself gave us were more complete.
  152. find_library(PYTHON_LIBRARY
  153. NAMES "python${PYTHON_LIBRARY_SUFFIX}"
  154. PATHS ${_PYTHON_LIBS_SEARCH}
  155. NO_DEFAULT_PATH)
  156. # If all else fails, just set the name/version and let the linker figure out the path.
  157. if(NOT PYTHON_LIBRARY)
  158. set(PYTHON_LIBRARY python${PYTHON_LIBRARY_SUFFIX})
  159. endif()
  160. endif()
  161. MARK_AS_ADVANCED(
  162. PYTHON_LIBRARY
  163. PYTHON_INCLUDE_DIR
  164. )
  165. # We use PYTHON_INCLUDE_DIR, PYTHON_LIBRARY and PYTHON_DEBUG_LIBRARY for the
  166. # cache entries because they are meant to specify the location of a single
  167. # library. We now set the variables listed by the documentation for this
  168. # module.
  169. SET(PYTHON_INCLUDE_DIRS "${PYTHON_INCLUDE_DIR}")
  170. SET(PYTHON_LIBRARIES "${PYTHON_LIBRARY}")
  171. SET(PYTHON_DEBUG_LIBRARIES "${PYTHON_DEBUG_LIBRARY}")
  172. find_package_message(PYTHON
  173. "Found PythonLibs: ${PYTHON_LIBRARY}"
  174. "${PYTHON_EXECUTABLE}${PYTHON_VERSION}")
  175. set(PYTHONLIBS_FOUND TRUE)