Skip to content

Commit 6dadda5

Browse files
committed
fixed test error when using external absl. uses add_compile_definitions in cmake
1 parent ed7600b commit 6dadda5

File tree

3 files changed

+12
-21
lines changed

3 files changed

+12
-21
lines changed

src/CMakeLists.txt

Lines changed: 4 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,7 @@ if (SPM_ABSL_PROVIDER STREQUAL "module" OR SPM_ABSL_PROVIDER STREQUAL "package")
2020
list(APPEND SPM_LIBS absl::flags_parse)
2121
list(APPEND SPM_LIBS absl::log)
2222
list(APPEND SPM_LIBS absl::check)
23-
if (MSVC)
24-
add_definitions("/D_USE_EXTERNAL_ABSL")
25-
else()
26-
add_definitions("-D_USE_EXTERNAL_ABSL")
27-
endif()
23+
add_compile_definitions(_USE_EXTERNAL_ABSL)
2824
elseif (SPM_ABSL_PROVIDER STREQUAL "internal")
2925
set(ABSL_FLAGS_SRCS ${CMAKE_CURRENT_SOURCE_DIR}/../third_party/absl/flags/flag.cc)
3026
endif()
@@ -76,22 +72,12 @@ elseif (SPM_PROTOBUF_PROVIDER STREQUAL "package")
7672
protobuf_generate_cpp(SPM_MODEL_PROTO_SRCS SPM_MODEL_PROTO_HDRS sentencepiece_model.proto)
7773
set(PROTOBUF_LITE_SRCS "")
7874
include_directories(${PROTOBUF_INCLUDE_DIR})
79-
if (MSVC)
80-
add_definitions("/D_USE_EXTERNAL_PROTOBUF")
81-
else()
82-
add_definitions("-D_USE_EXTERNAL_PROTOBUF")
83-
endif()
75+
add_compile_definitions(_USE_EXTERNAL_PROTOBUF)
8476
endif()
8577

8678
include_directories(${CMAKE_CURRENT_BINARY_DIR})
8779
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/../third_party)
8880

89-
if (MSVC)
90-
add_definitions("/D_USE_INTERNAL_STRING_VIEW")
91-
else()
92-
add_definitions("-D_USE_INTERNAL_STRING_VIEW")
93-
endif()
94-
9581
set(SPM_SRCS
9682
${PROTOBUF_LITE_SRCS}
9783
${SPM_PROTO_HDRS}
@@ -187,12 +173,12 @@ list(APPEND SPM_LIBS ${PROTOBUF_LITE_LIBRARY} Threads::Threads)
187173
if (SPM_ENABLE_NFKC_COMPILE)
188174
find_package(ICU 4.4 COMPONENTS i18n data uc REQUIRED)
189175
include_directories(${ICU_INCLUDE_DIRS})
190-
add_definitions(-DENABLE_NFKC_COMPILE)
176+
add_compile_definitions(ENABLE_NFKC_COMPILE)
191177
list(APPEND SPM_LIBS ICU::i18n ICU::data ICU::uc)
192178
endif()
193179

194180
if (SPM_DISABLE_EMBEDDED_DATA)
195-
add_definitions(-DDISABLE_EMBEDDED_DATA)
181+
add_compile_definitions(DISABLE_EMBEDDED_DATA)
196182
endif()
197183

198184
if (SPM_ENABLE_TCMALLOC)

src/init_test.cc

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717
#include "common.h"
1818
#include "testharness.h"
1919

20+
#ifndef _USE_EXTERNAL_ABSL
21+
2022
ABSL_FLAG(int32_t, int32_f, 10, "int32_flags");
2123
ABSL_FLAG(bool, bool_f, false, "bool_flags");
2224
ABSL_FLAG(int64_t, int64_f, 9223372036854775807LL, "int64_flags");
@@ -87,8 +89,6 @@ TEST(FlagsTest, ParseCommandLineFlagsTest3) {
8789
EXPECT_EQ(1, argc);
8890
}
8991

90-
#ifndef _USE_EXTERNAL_ABSL
91-
9292
TEST(FlagsTest, ParseCommandLineFlagsHelpTest) {
9393
const char *kFlags[] = {"program", "--help"};
9494
int argc = arraysize(kFlags);
@@ -143,5 +143,6 @@ TEST(FlagsTest, ParseCommandLineFlagsEmptyIntArgs) {
143143
char **argv = const_cast<char **>(kFlags);
144144
EXPECT_DEATH(ParseCommandLineFlags(kFlags[0], &argc, &argv), );
145145
}
146-
#endif // _USE_EXTERNAL_ABSL
147146
} // namespace absl
147+
148+
#endif // _USE_EXTERNAL_ABSL

src/sentencepiece_processor.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,15 @@
2323
#include <utility>
2424
#include <vector>
2525

26+
#ifdef _USE_EXTERNAL_ABSL
27+
#include "third_party/absl/strings/string_view.h"
28+
#else // _USE_EXTERNAL_ABSL
2629
#ifndef SWIG
2730
namespace absl {
2831
using std::string_view;
2932
} // namespace absl
3033
#endif // SWIG
34+
#endif // _USE_EXTERNAL_ABSL
3135

3236
namespace sentencepiece {
3337
namespace util {

0 commit comments

Comments
 (0)