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.

tvm_gpu.cmake 995 B

12345678910111213141516
  1. set(incubator_tvm_gpu_CFLAGS "-pipe -Wall -fPIC -fstack-protector-all -D_FORTIFY_SOURCE=2 -O2")
  2. set(incubator_tvm_gpu_CXXFLAGS "-std=c++11 -pipe -Wall -fPIC -fstack-protector-all -D_FORTIFY_SOURCE=2 -O2")
  3. set(USE_CUDA "ON")
  4. mindspore_add_pkg(incubator_tvm_gpu
  5. VER 0.6.0
  6. LIBS tvm
  7. URL https://github.com/apache/incubator-tvm/archive/v0.6.0.tar.gz
  8. MD5 9cbbd32545a776023acabbba270449fe
  9. SUBMODULES ${dlpack_DIRPATH} ${dmlc-core_DIRPATH} ${rang_DIRPATH}
  10. SOURCEMODULES topi/python/topi python/tvm
  11. PATCHES ${CMAKE_SOURCE_DIR}/third_party/patch/incubator-tvm/find_library.patch
  12. ${CMAKE_SOURCE_DIR}/third_party/patch/incubator-tvm/include.patch
  13. ${CMAKE_SOURCE_DIR}/third_party/patch/incubator-tvm/src_pass.patch
  14. CMAKE_OPTION -DBUILD_TESTING=OFF -DCMAKE_POSITION_INDEPENDENT_CODE=ON -DBUILD_SHARED_LIBS=ON)
  15. include_directories(${incubator_tvm_gpu_INC})
  16. add_library(mindspore::tvm ALIAS incubator_tvm_gpu::tvm)