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