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.

sentencepiece.cmake 2.0 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. if(ENABLE_GITEE)
  2. set(REQ_URL "https://gitee.com/mirrors/sentencepiece/repository/archive/v0.1.92.tar.gz")
  3. set(MD5 "618f5590c99884866c01cb773096c523")
  4. else()
  5. set(REQ_URL "https://github.com/google/sentencepiece/archive/v0.1.92.tar.gz")
  6. set(MD5 "5dfd2241914b5598a68b2a8542ed8e91")
  7. endif()
  8. if(WIN32)
  9. set(sentencepiece_CXXFLAGS "-D_FORTIFY_SOURCE=2 -O2 -Wno-unused-result -Wno-stringop-overflow \
  10. -Wno-format-extra-args -Wno-format")
  11. set(sentencepiece_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
  12. mindspore_add_pkg(sentencepiece
  13. VER 0.1.92
  14. LIBS sentencepiece sentencepiece_train
  15. URL ${REQ_URL}
  16. CMAKE_OPTION -DCMAKE_BUILD_TYPE=Release -DSPM_USE_BUILTIN_PROTOBUF=ON
  17. MD5 ${MD5}
  18. )
  19. else()
  20. set(sentencepiece_CXXFLAGS "-D_FORTIFY_SOURCE=2 -O2 -Wno-unused-result -Wno-sign-compare")
  21. set(sentencepiece_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
  22. if(ENABLE_GLIBCXX)
  23. mindspore_add_pkg(sentencepiece
  24. VER 0.1.92
  25. LIBS sentencepiece sentencepiece_train
  26. URL ${REQ_URL}
  27. CMAKE_OPTION -DCMAKE_BUILD_TYPE=Release -DSPM_USE_BUILTIN_PROTOBUF=OFF -DSPM_ENABLE_SHARED=OFF
  28. -DPROTOBUF_INC=${protobuf_INC}
  29. MD5 ${MD5}
  30. PATCHES ${CMAKE_SOURCE_DIR}/third_party/patch/sentencepiece/sentencepiece.patch001_cpu
  31. )
  32. else()
  33. mindspore_add_pkg(sentencepiece
  34. VER 0.1.92
  35. LIBS sentencepiece sentencepiece_train
  36. URL ${REQ_URL}
  37. CMAKE_OPTION -DCMAKE_BUILD_TYPE=Release -DSPM_USE_BUILTIN_PROTOBUF=OFF -DSPM_ENABLE_SHARED=OFF
  38. -DPROTOBUF_INC=${protobuf_INC}
  39. MD5 ${MD5}
  40. PATCHES ${CMAKE_SOURCE_DIR}/third_party/patch/sentencepiece/sentencepiece.patch001
  41. )
  42. endif()
  43. endif()
  44. include_directories(${sentencepiece_INC})
  45. add_library(mindspore::sentencepiece ALIAS sentencepiece::sentencepiece)
  46. add_library(mindspore::sentencepiece_train ALIAS sentencepiece::sentencepiece_train)