Huawei_Technology
/
MindSpore-AKG
Not watched
1
0
Fork 0
Code
Releases 1 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
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.
Tree: a5a856cd2c
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'a5a856cd2c'
${ noResults }
MindSpore-AKG/CMakeLists.txt

323 lines
12 kB
Raw Blame History 

  1. # Copyright 2020-2021 Huawei Technologies Co., Ltd

  2. #

  3. # Licensed under the Apache License, Version 2.0 (the "License");

  4. # you may not use this file except in compliance with the License.

  5. # You may obtain a copy of the License at

  6. #

  7. # http://www.apache.org/licenses/LICENSE-2.0

  8. #

  9. # Unless required by applicable law or agreed to in writing, software

  10. # distributed under the License is distributed on an "AS IS" BASIS,

  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  12. # See the License for the specific language governing permissions and

  13. # limitations under the License.

  14. 

  15. cmake_minimum_required(VERSION 3.14)

  16. 

  17. if(POLICY CMP0054)

  18.   cmake_policy(SET CMP0054 NEW)

  19. endif()

  20. if(POLICY CMP0072)

  21.   cmake_policy(SET CMP0072 NEW)

  22. endif()

  23. 

  24. project(akg C CXX)

  25. 

  26. set(AKG_SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}")

  27. set(TVM_DIR "${AKG_SOURCE_DIR}/third_party/incubator-tvm")

  28. 

  29. if(NOT CMAKE_SYSTEM_NAME MATCHES "Windows" AND NOT USE_KC_AIR AND NOT USE_CCE_PROFILING)

  30.   add_compile_definitions(_GLIBCXX_USE_CXX11_ABI=0)

  31. endif()

  32. 

  33. # Find gmp

  34. find_path(GMP_INCLUDE_DIR NAME gmp.h HINTS /usr /usr/local PATH_SUFFIXES include include/x86_64-linux-gnu)

  35. find_library(GMP_LIBRARY NAMES gmp HINTS /usr /usr/local PATH_SUFFIXES lib lib64 lib/x86_64-linux-gnu)

  36. if(NOT GMP_INCLUDE_DIR)

  37.   message(FATAL_ERROR "Please export CMAKE_INCLUDE_PATH to directory where gmp.h locates at.")

  38. endif()

  39. if(NOT GMP_LIBRARY)

  40.   message(FATAL_ERROR "Please export CMAKE_LIBRARY_PATH to directory where libgmp.so locates at.")

  41. endif()

  42. 

  43. include(cmake/RT.cmake)

  44. include(cmake/utils.cmake)

  45. include(cmake/external_libs/isl.cmake)

  46. 

  47. if(USE_CCE_RT)

  48.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/nnae/latest/fwkacllib/lib64)

  49.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64)

  50.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/fwkacllib/lib64)

  51.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/nnae/latest/fwkacllib/lib64/plugin/opskernel)

  52.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64/plugin/opskernel)

  53.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/fwkacllib/lib64/plugin/opskernel)

  54.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/add-ons)

  55.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  56.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  57.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)

  58. elseif(USE_CUDA)

  59.   set(AKG_RPATH ${AKG_RPATH}:/usr/local/cuda/lib64)

  60. endif()

  61. link_directories(${AKG_RPATH})

  62. 

  63. # Search AKG_EXTEND by order

  64. set(AKG_EXTEND )

  65. if(NOT USE_CUDA)

  66.   set(AKG_EXTEND_FILE )

  67.   set(LIB_PATH1 ${AKG_SOURCE_DIR}/libakg_ext.a)

  68.   set(LIB_PATH2 ${AKG_SOURCE_DIR}/build/libakg_ext.a)

  69. 

  70.   if(EXISTS ${LIB_PATH1})  # Search libakg_ext.a in akg/

  71.     set(AKG_EXTEND_FILE ${LIB_PATH1})

  72.   else()

  73.     if(EXISTS ${LIB_PATH2})  # Search libakg_ext.a in akg/build/

  74.       set(AKG_EXTEND_FILE ${LIB_PATH2})

  75.     elseif(NOT USE_KC_AIR AND NOT USE_CCE_PROFILING)  # Download libakg_ext.a to akg/build/

  76.       execute_process(COMMAND bash ${AKG_SOURCE_DIR}/build.sh -a

  77.               WORKING_DIRECTORY ${AKG_SOURCE_DIR}

  78.               OUTPUT_VARIABLE OUTPUT_URL

  79.               RESULT_VARIABLE RESULT)

  80.       if(RESULT EQUAL 0 AND OUTPUT_URL MATCHES "libakg_ext.a")

  81.         # Download library

  82.         string(STRIP ${OUTPUT_URL} LIB_URL)

  83.         message("-- Downloading ${LIB_URL} --> ${LIB_PATH2}")

  84.         file(DOWNLOAD ${LIB_URL} ${LIB_PATH2} STATUS DOWNLOAD_STATUS)

  85.         message("-- Download status: ${DOWNLOAD_STATUS}")

  86.         list(GET DOWNLOAD_STATUS 0 DOWNLOAD_CODE)

  87.         if(DOWNLOAD_CODE EQUAL 0)

  88.           set(AKG_EXTEND_FILE ${LIB_PATH2})

  89.         endif()

  90.       endif()

  91.     endif()

  92.   endif()

  93. 

  94.   message("-- AKG_EXTEND_FILE: ${AKG_EXTEND_FILE}")

  95.   if(EXISTS ${AKG_EXTEND_FILE})

  96.     file(COPY ${AKG_EXTEND_FILE} DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/akg_extend)

  97.     execute_process(COMMAND ar -x libakg_ext.a

  98.             WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/akg_extend)

  99.     file(GLOB AKG_EXTEND ${CMAKE_CURRENT_BINARY_DIR}/akg_extend/*.o)

  100.   endif()

  101. endif()

  102. if(NOT AKG_EXTEND)

  103.   message("-- Build without AKG_EXTEND support")

  104. else()

  105.   message("-- Build with AKG_EXTEND support")

  106. endif()

  107. 

  108. file(COPY ${AKG_SOURCE_DIR}/python/akg DESTINATION

  109.     ${CMAKE_CURRENT_BINARY_DIR})

  110. 

  111. # Utility functions

  112. include(${TVM_DIR}/cmake/util/Util.cmake)

  113. 

  114. tvm_option(USE_ASAN "Build with AddressSanitizer" OFF)

  115. tvm_option(USE_CUDA "Build with CUDA" OFF)

  116. tvm_option(USE_CUDNN "Build with cuDNN" OFF)

  117. tvm_option(USE_LLVM "Build with LLVM" OFF)

  118. 

  119. 

  120. tvm_option(

  121.   USE_DEFAULT_LOG

  122.   "Use Customize log to eliminate useless log. If you want to enable defalut log, set this option to ON"

  123.   OFF)

  124. 

  125. set(CMAKE_EXPORT_COMPILE_COMMANDS ON)

  126. # include directories

  127. include_directories(AFTER "${TVM_DIR}/include")

  128. include_directories(AFTER "${TVM_DIR}")

  129. include_directories(AFTER "${TVM_DIR}/src")

  130. include_directories(AFTER "${TVM_DIR}/src/schedule")

  131. include_directories(AFTER "${AKG_SOURCE_DIR}/src")

  132. include_directories(AFTER "${AKG_SOURCE_DIR}/src/include")

  133. include_directories(AFTER "${TVM_DIR}/3rdparty/dmlc-core/include")

  134. include_directories(AFTER "${TVM_DIR}/3rdparty/dlpack/include")

  135. include_directories(AFTER "${TVM_DIR}/3rdparty/compiler-rt")

  136. include_directories(AFTER "${TVM_DIR}/3rdparty/rang/include")

  137. include_directories(AFTER "${TVM_DIR}/3rdparty/picojson")

  138. # cce rt

  139. include_directories(AFTER "${AKG_SOURCE_DIR}/third_party/fwkacllib/inc")

  140. include_directories(AFTER "${AKG_SOURCE_DIR}/third_party/fwkacllib/inc/toolchain")

  141. 

  142. # initial variables

  143. set(TVM_LINKER_LIBS )

  144. 

  145. add_definitions(-DPICOJSON_USE_INT64=1)

  146. add_definitions(-DDMLC_LOG_CUSTOMIZE=1)

  147. if(USE_AKG_LOG)

  148.   add_definitions(-DUSE_AKG_LOG=1)

  149. endif()

  150. if(NOT AKG_EXTEND)

  151.   add_definitions(-DUSE_AKG_COMPILE_STUB=1)

  152. endif()

  153. if(USE_CCE_PROFILING)

  154.   add_definitions(-DUSE_CCE_PROFILING=1)

  155. endif()

  156. 

  157. # Generic compilation options

  158. include(CheckCXXCompilerFlag)

  159. check_cxx_compiler_flag("-std=c++11" SUPPORT_CXX11)

  160. if(NOT SUPPORT_CXX11)

  161.   message(

  162.     FATAL_ERROR "-- please choose a compiler which support C++ 11 standard")

  163. endif()

  164. 

  165. check_cxx_compiler_flag("-march=native" NATIVE_BUILD)

  166. if(NATIVE_BUILD AND ENABLE_NATIVE_BUILD)

  167.   set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -march=native")

  168.   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -march=native")

  169. endif()

  170. 

  171. set(CMAKE_SKIP_RPATH TRUE)

  172. set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -pipe -Wall -fPIC -fstack-protector-all")

  173. set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -Wl,-z,relro,-z,now,-z,noexecstack")

  174. 

  175. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -pipe -Wall -fPIC -fstack-protector-all")

  176. set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wl,-z,relro,-z,now,-z,noexecstack")

  177. 

  178. if("${CMAKE_BUILD_TYPE}" STREQUAL "Debug")

  179.   message("-- Build in Debug mode")

  180.   set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -O0 -g -rdynamic")

  181.   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O0 -g -rdynamic")

  182. else()

  183.   message("-- Build in Release mode")

  184.   set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -O2 -D_FORTIFY_SOURCE=2 -Werror")

  185.   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O2 -D_FORTIFY_SOURCE=2 -Werror")

  186. endif()

  187. if(CMAKE_CXX_COMPILER_ID MATCHES "GNU" AND CMAKE_CXX_COMPILER_VERSION

  188.                                            VERSION_GREATER 7.0)

  189.   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -faligned-new")

  190. endif()

  191. if(USE_ASAN)

  192.   set(CMAKE_C_FLAGS

  193.       "-fsanitize=address -fno-omit-frame-pointer  ${CMAKE_C_FLAGS}")

  194.   set(CMAKE_CXX_FLAGS

  195.       "-fsanitize=address -fno-omit-frame-pointer  ${CMAKE_CXX_FLAGS}")

  196. endif()

  197. 

  198. # add source group

  199. file(GLOB_RECURSE GROUP_SOURCE "${TVM_DIR}/src/*.cc" "src/*.cc")

  200. file(GLOB_RECURSE GROUP_INCLUDE "${TVM_DIR}/src/*.h"

  201.      "${TVM_DIR}/include/*.h" "src/*.h" "include/*.h")

  202. assign_source_group("Source" ${GROUP_SOURCE})

  203. assign_source_group("Include" ${GROUP_INCLUDE})

  204. 

  205. # Source file lists

  206. file(

  207.   GLOB

  208.   COMPILER_SRCS

  209.   ${TVM_DIR}/3rdparty/dmlc-core/src/io/filesys.cc

  210.   ${TVM_DIR}/3rdparty/dmlc-core/src/io/indexed_recordio_split.cc

  211.   ${TVM_DIR}/3rdparty/dmlc-core/src/io/input_split_base.cc

  212.   ${TVM_DIR}/3rdparty/dmlc-core/src/io/line_split.cc

  213.   ${TVM_DIR}/3rdparty/dmlc-core/src/io/recordio_split.cc

  214.   ${TVM_DIR}/3rdparty/dmlc-core/src/io/local_filesys.cc

  215.   ${TVM_DIR}/3rdparty/dmlc-core/src/io.cc

  216.   ${TVM_DIR}/src/api/*.cc

  217.   ${TVM_DIR}/src/arithmetic/*.cc

  218.   ${TVM_DIR}/src/codegen/*.cc

  219.   ${TVM_DIR}/src/lang/*.cc

  220.   ${TVM_DIR}/src/pass/*.cc

  221.   ${TVM_DIR}/src/op/*.cc

  222.   ${TVM_DIR}/src/node/*.cc

  223.   ${TVM_DIR}/src/schedule/*.cc

  224.   ${TVM_DIR}/src/runtime/*.cc

  225.   ${TVM_DIR}/src/runtime/cce/*.cc

  226.   ${TVM_DIR}/src/runtime/vm/*.cc

  227.   ${TVM_DIR}/src/runtime/vm/profiler/*.cc

  228.   ${TVM_DIR}/src/codegen/stackvm/*.cc

  229.   ${AKG_SOURCE_DIR}/src/poly/*.cc

  230.   ${AKG_SOURCE_DIR}/src/poly/schedule_pass/*.cc

  231.   ${AKG_SOURCE_DIR}/src/poly/schedule_pass_gpu/*.cc

  232.   ${AKG_SOURCE_DIR}/src/poly/tiling/*.cc

  233.   ${AKG_SOURCE_DIR}/src/poly/gpu_emit/*.cc

  234.   ${AKG_SOURCE_DIR}/src/api/*.cc

  235.   ${AKG_SOURCE_DIR}/src/pass/*.cc

  236.   ${AKG_SOURCE_DIR}/src/schedule/*.cc

  237.   ${AKG_SOURCE_DIR}/src/codegen/*.cc

  238.   ${AKG_SOURCE_DIR}/src/composite/*.cc

  239.   ${AKG_SOURCE_DIR}/src/composite/optimize/*.cc

  240.   ${AKG_SOURCE_DIR}/src/common/*.cc

  241.   ${AKG_SOURCE_DIR}/src/runtime/stub/*.cc)

  242. 

  243. file(GLOB RUNTIME_STUB_SRC ${AKG_SOURCE_DIR}/src/runtime/stub/*.cc)

  244. if(USE_CCE_RT OR USE_KC_AIR)

  245.   list(REMOVE_ITEM COMPILER_SRCS ${RUNTIME_STUB_SRC})

  246. endif()

  247. 

  248. 

  249. file(GLOB DATATYPE_SRCS ${TVM_DIR}/src/codegen/datatype/*.cc)

  250. list(APPEND COMPILER_SRCS ${DATATYPE_SRCS})

  251. 

  252. file(GLOB TOPI_SRCS ${TVM_DIR}/topi/src/*.cc)

  253. 

  254. file(GLOB_RECURSE RELAY_SRCS ${TVM_DIR}/src/relay/*.cc)

  255. list(APPEND COMPILER_SRCS ${RELAY_SRCS})

  256. 

  257. file(GLOB COMPILER_VERILOG_SRCS ${TVM_DIR}/src/codegen/veriog/*.cc)

  258. list(APPEND COMPILER_SRCS ${COMPILER_VERILOG_SRCS})

  259. 

  260. file(

  261.   GLOB

  262.   RUNTIME_SRCS

  263.   ${TVM_DIR}/src/runtime/*.cc

  264.   ${TVM_DIR}/src/runtime/vm/*.cc

  265.   ${TVM_DIR}/src/runtime/stackvm/*.cc)

  266. 

  267. if(USE_CUDA)

  268.   include(${TVM_DIR}/cmake/util/FindCUDA.cmake)

  269.   include(${TVM_DIR}/cmake/modules/CUDA.cmake)

  270.   file(GLOB PROFILER_SRCS ${AKG_SOURCE_DIR}/src/profiler/*.cc)

  271.   list(APPEND COMPILER_SRCS ${PROFILER_SRCS})

  272. endif()

  273. 

  274. if(USE_LLVM)

  275.   include(${TVM_DIR}/cmake/util/FindLLVM.cmake)

  276.   file(GLOB LLVM_SRCS ${TVM_DIR}/src/codegen/llvm/*.cc)

  277.   list(APPEND COMPILER_SRCS ${LLVM_SRCS})

  278.   include(${TVM_DIR}/cmake/modules/LLVM.cmake)

  279. endif()

  280. 

  281. if(USE_RPC)

  282.   message(STATUS "Build with RPC support...")

  283.   file(GLOB RUNTIME_RPC_SRCS ${TVM_DIR}/src/runtime/rpc/*.cc)

  284.   list(APPEND RUNTIME_SRCS ${RUNTIME_RPC_SRCS})

  285. endif(USE_RPC)

  286. 

  287. file(GLOB COMPILER_OFF_SRC ${TVM_DIR}/src/codegen/opt/build_*_off.cc)

  288. list(APPEND COMPILER_SRCS ${COMPILER_OFF_SRC})

  289. 

  290. if(USE_CUDA)

  291.   list(REMOVE_ITEM COMPILER_SRCS ${TVM_DIR}/src/codegen/opt/build_cuda_off.cc)

  292. endif(USE_CUDA)

  293. 

  294. add_library(akg SHARED ${COMPILER_SRCS} ${RUNTIME_SRCS} ${TOPI_SRCS} ${AKG_EXTEND})

  295. 

  296. add_dependencies(akg akg::isl_fixed)

  297. target_link_libraries(akg ${TVM_LINKER_LIBS} ${TVM_RUNTIME_LINKER_LIBS} akg::isl_fixed ${GMP_LIBRARY} pthread)

  298. 

  299. # Related headers

  300. target_include_directories(akg PRIVATE "${TVM_DIR}/topi/include")

  301. 

  302. # Installation rules

  303. if(ENABLE_AKG)

  304.   install(TARGETS akg DESTINATION lib${LIB_SUFFIX})

  305. else()

  306.   install(TARGETS akg DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/akg/lib)

  307. endif()

  308. 

  309. install(

  310.   DIRECTORY

  311.   ${TVM_DIR}/python/tvm

  312.   ${TVM_DIR}/topi/python/topi

  313.   DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/akg)

  314. install(

  315.   DIRECTORY

  316.   ${AKG_SOURCE_DIR}/src/akg_reduce

  317.   ${AKG_SOURCE_DIR}/src/paris_reduce

  318.   ${AKG_SOURCE_DIR}/src/akg_mma_lib

  319.   DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/akg/include)

  320. file(GLOB REPOSITORY_FILE_LIST ${AKG_SOURCE_DIR}/python/akg/composite/*.json)

  321. install(FILES ${REPOSITORY_FILE_LIST} DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/akg/config)