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.7 kB

5 years ago
5 years ago
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  1. cmake_minimum_required(VERSION 3.14)
  2. project (MindSpore)
  3. include(${CMAKE_SOURCE_DIR}/cmake/options.cmake)
  4. set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CMAKE_SOURCE_DIR}/cmake/modules/")
  5. if (${CMAKE_SYSTEM_NAME} MATCHES "Darwin")
  6. set(CMAKE_CXX_FLAGS_RELEASE "$ENV{CXXFLAGS} -O2 -Werror -Wno-return-std-move -Wno-unused-private-field -Wno-unused-lambda-capture -Wno-sign-compare -Wno-overloaded-virtual -Wno-unneeded-internal-declaration -Wno-unused-variable -Wno-pessimizing-move -Wno-inconsistent-missing-override -DHALF_ENABLE_CPP11_USER_LITERALS=0 -D_FORTIFY_SOURCE=2")
  7. else()
  8. set(CMAKE_CXX_FLAGS_RELEASE "$ENV{CXXFLAGS} -O2 -Wl,--allow-shlib-undefined -DHALF_ENABLE_CPP11_USER_LITERALS=0 -D_FORTIFY_SOURCE=2")
  9. endif()
  10. set(CMAKE_CXX_FLAGS_DEBUG "$ENV{CXXFLAGS} -O0 -g2 -ggdb -fno-inline-functions -fno-omit-frame-pointer -Wl,--allow-shlib-undefined -D_LIBCPP_INLINE_VISIBILITY='' -D'_LIBCPP_EXTERN_TEMPLATE(...)=' -DHALF_ENABLE_CPP11_USER_LITERALS=0 -D_FORTIFY_SOURCE=2 -Wno-cpp")
  11. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -I/usr/local/include -std=c++17 -Werror -Wall -Wno-deprecated-declarations -fPIC")
  12. set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
  13. set(PYBIND11_CPP_STANDARD -std=c++17)
  14. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OPTION_CXX_FLAGS}")
  15. find_package(Threads)
  16. find_package(Patch)
  17. message(PATCH_EXECUTABLE = ${Patch_EXECUTABLE})
  18. include(${CMAKE_SOURCE_DIR}/cmake/mind_expression.cmake)
  19. include_directories(${CMAKE_CURRENT_SOURCE_DIR})
  20. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include)
  21. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/third_party/flatbuffers/include/flatbuffers)
  22. include(${CMAKE_SOURCE_DIR}/cmake/dependency_utils.cmake)
  23. find_package(Python3 3.7 COMPONENTS Interpreter Development)
  24. if(Python3_FOUND)
  25. set(PYTHON_INCLUDE_DIRS "${Python3_INCLUDE_DIRS}")
  26. set(PYTHON_LIBRARIES "${Python3_LIBRARIES}")
  27. if (WIN32)
  28. if (Python3_DIR)
  29. message("Python3_DIR set already: " ${Python3_DIR})
  30. else()
  31. string(LENGTH ${PYTHON_LIBRARIES} PYTHON_LIBRARIES_LEN)
  32. string(LENGTH "libpythonxx.a" Python3_NAME_LEN)
  33. math(EXPR Python3_DIR_LEN ${PYTHON_LIBRARIES_LEN}-${Python3_NAME_LEN})
  34. string(SUBSTRING ${Python3_LIBRARIES} 0 ${Python3_DIR_LEN} Python3_DIR)
  35. message("Python3_DIR: " ${Python3_DIR})
  36. endif()
  37. link_directories(${Python3_DIR})
  38. endif()
  39. else()
  40. find_python_package(py_inc py_lib)
  41. set(PYTHON_INCLUDE_DIRS "${py_inc}")
  42. set(PYTHON_LIBRARIES "${py_lib}")
  43. endif()
  44. message("PYTHON_INCLUDE_DIRS = ${PYTHON_INCLUDE_DIRS}")
  45. message("PYTHON_LIBRARIES = ${PYTHON_LIBRARIES}")
  46. include_directories(${PYTHON_INCLUDE_DIRS})
  47. set(MS_CCSRC_PATH ${CMAKE_SOURCE_DIR}/mindspore/ccsrc)
  48. set(MS_CCSRC_BUILD_PATH ${BUILD_PATH}/mindspore/mindspore/ccsrc)
  49. if (ENABLE_GE)
  50. link_directories(${CMAKE_SOURCE_DIR}/third_party/ge/lib)
  51. elseif(ENABLE_D OR ENABLE_TESTCASES)
  52. include(${CMAKE_SOURCE_DIR}/cmake/dependency_graphengine.cmake)
  53. endif()
  54. if (ENABLE_GE OR ENABLE_D OR ENABLE_TESTCASES)
  55. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/graphengine/inc)
  56. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/graphengine/inc/external)
  57. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/graphengine/inc/framework)
  58. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/graphengine/third_party/fwkacllib/inc)
  59. include_directories(${CMAKE_CURRENT_SOURCE_DIR}/graphengine/third_party/fwkacllib/inc/toolchain)
  60. endif()
  61. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fvisibility=hidden")
  62. add_subdirectory(mindspore/ccsrc)
  63. if (ENABLE_TESTCASES)
  64. add_subdirectory(tests)
  65. endif()
  66. include(cmake/package.cmake)