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 820 B

123456789101112131415
  1. set(incubator_tvm_gpu_CXXFLAGS "-D_FORTIFY_SOURCE=2 -O2")
  2. set(incubator_tvm_gpu_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
  3. mindspore_add_pkg(incubator_tvm_gpu
  4. VER 0.6.0
  5. LIBS tvm
  6. URL https://github.com/apache/incubator-tvm/archive/v0.6.0.tar.gz
  7. MD5 9cbbd32545a776023acabbba270449fe
  8. CUSTOM_CMAKE ${CMAKE_SOURCE_DIR}/third_party/patch/incubator-tvm/
  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 " ")
  15. add_library(mindspore::tvm ALIAS incubator_tvm_gpu::tvm)