You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

CMakeLists.txt 3.9 kB

5 years ago
5 years ago
5 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  1. cmake_minimum_required(VERSION 3.14.0)
  2. project(MindSpore)
  3. if(CMAKE_CXX_COMPILER_ID STREQUAL "GNU" AND CMAKE_CXX_COMPILER_VERSION VERSION_LESS 7.3.0)
  4. message(FATAL_ERROR "GCC version ${CMAKE_CXX_COMPILER_VERSION} must not be less than 7.3.0")
  5. endif()
  6. include(${CMAKE_SOURCE_DIR}/cmake/options.cmake)
  7. include(${CMAKE_SOURCE_DIR}/cmake/check_requirements.cmake)
  8. set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/cmake/modules/")
  9. if(NOT CMAKE_SYSTEM_NAME MATCHES "Windows")
  10. if(NOT ENABLE_GLIBCXX)
  11. add_compile_definitions(_GLIBCXX_USE_CXX11_ABI=0)
  12. endif()
  13. endif()
  14. if(${CMAKE_SYSTEM_NAME} MATCHES "Darwin")
  15. set(CMAKE_OSX_SYSROOT "")
  16. set(CMAKE_CXX_FLAGS_RELEASE "$ENV{CXXFLAGS} -O2 -Winconsistent-missing-override -Wuser-defined-warnings \
  17. -Wno-return-std-move -Wno-unused-private-field -Wno-unused-lambda-capture -Wno-sign-compare \
  18. -Wno-overloaded-virtual -Wno-unneeded-internal-declaration -Wno-unused-variable -Wno-pessimizing-move \
  19. -Wno-inconsistent-missing-override -DHALF_ENABLE_CPP11_USER_LITERALS=0 -D_FORTIFY_SOURCE=2")
  20. else()
  21. set(CMAKE_CXX_FLAGS_RELEASE "$ENV{CXXFLAGS} -O2 -Wl,--allow-shlib-undefined \
  22. -DHALF_ENABLE_CPP11_USER_LITERALS=0 -D_FORTIFY_SOURCE=2")
  23. endif()
  24. if(ENABLE_PYTHON)
  25. add_compile_definitions(ENABLE_PYTHON)
  26. endif()
  27. set(CMAKE_CXX_FLAGS_DEBUG "$ENV{CXXFLAGS} -O0 -g2 -ggdb -fno-inline-functions -fno-omit-frame-pointer \
  28. -Wl,--allow-shlib-undefined -D_LIBCPP_INLINE_VISIBILITY='' -D_LIBCPP_DISABLE_EXTERN_TEMPLATE=1 \
  29. -DHALF_ENABLE_CPP11_USER_LITERALS=0 -D_FORTIFY_SOURCE=2 -Wno-cpp")
  30. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -I/usr/local/include -std=c++17 \
  31. -Werror -Wall -Wno-deprecated-declarations -fPIC")
  32. set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
  33. set(PYBIND11_CPP_STANDARD -std=c++17)
  34. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OPTION_CXX_FLAGS}")
  35. if(ENABLE_AKG AND (ENABLE_D OR ENABLE_GPU))
  36. add_subdirectory("${CMAKE_SOURCE_DIR}/akg")
  37. endif()
  38. include(${CMAKE_SOURCE_DIR}/cmake/mind_expression.cmake)
  39. include_directories(${CMAKE_CURRENT_SOURCE_DIR})
  40. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/third_party/securec/include)
  41. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include)
  42. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers)
  43. include(${CMAKE_SOURCE_DIR}/cmake/dependency_utils.cmake)
  44. find_package(Python3 COMPONENTS Interpreter Development)
  45. if(Python3_FOUND)
  46. set(PYTHON_INCLUDE_DIRS "${Python3_INCLUDE_DIRS}")
  47. set(PYTHON_LIBRARIES "${Python3_LIBRARIES}")
  48. if(WIN32)
  49. if(Python3_DIR)
  50. message("Python3_DIR set already: " ${Python3_DIR})
  51. else()
  52. string(LENGTH ${PYTHON_LIBRARIES} PYTHON_LIBRARIES_LEN)
  53. string(LENGTH "libpythonxx.a" Python3_NAME_LEN)
  54. math(EXPR Python3_DIR_LEN ${PYTHON_LIBRARIES_LEN}-${Python3_NAME_LEN})
  55. string(SUBSTRING ${Python3_LIBRARIES} 0 ${Python3_DIR_LEN} Python3_DIR)
  56. message("Python3_DIR: " ${Python3_DIR})
  57. endif()
  58. link_directories(${Python3_DIR})
  59. endif()
  60. else()
  61. find_python_package(py_inc py_lib)
  62. set(PYTHON_INCLUDE_DIRS "${py_inc}")
  63. set(PYTHON_LIBRARIES "${py_lib}")
  64. endif()
  65. message("PYTHON_INCLUDE_DIRS = ${PYTHON_INCLUDE_DIRS}")
  66. message("PYTHON_LIBRARIES = ${PYTHON_LIBRARIES}")
  67. include_directories(${PYTHON_INCLUDE_DIRS})
  68. set(MS_CCSRC_PATH ${CMAKE_SOURCE_DIR}/mindspore/ccsrc)
  69. set(MS_CCSRC_BUILD_PATH ${BUILD_PATH}/mindspore/mindspore/ccsrc)
  70. if(ENABLE_D OR ENABLE_ACL OR ENABLE_TESTCASES)
  71. include(${CMAKE_SOURCE_DIR}/cmake/dependency_graphengine.cmake)
  72. endif()
  73. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden")
  74. add_subdirectory(mindspore/ccsrc)
  75. add_subdirectory(mindspore/core)
  76. if(ENABLE_TESTCASES OR ENABLE_CPP_ST)
  77. add_subdirectory(tests)
  78. endif()
  79. include(cmake/package.cmake)