Skip to content

Commit

Permalink
Detect and print build type. Refs #88
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 614873909
  • Loading branch information
jan-wassenberg authored and Copybara-Service committed Mar 12, 2024
1 parent a9be065 commit e760948
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 2 deletions.
4 changes: 2 additions & 2 deletions run.cc
Expand Up @@ -66,8 +66,8 @@ void ShowConfig(LoaderArgs& loader, InferenceArgs& inference, AppArgs& app) {
<< std::thread::hardware_concurrency() << std::endl
<< "Instruction set : "
<< hwy::TargetName(hwy::DispatchedTarget()) << " ("
<< hwy::VectorBytes() * 8 << " bits)"
<< "\n"
<< hwy::VectorBytes() * 8 << " bits)" << "\n"
<< "Compiled config : " << CompiledConfig() << "\n"
<< "Weight Type : "
<< gcpp::TypeName(gcpp::WeightT()) << "\n"
<< "EmbedderInput Type : "
Expand Down
18 changes: 18 additions & 0 deletions util/app.h
Expand Up @@ -36,6 +36,24 @@

namespace gcpp {

static inline const char* CompiledConfig() {
#if HWY_IS_ASAN
return "asan";
#elif HWY_IS_MSAN
return "msan";
#elif HWY_IS_TSAN
return "tsan";
#elif HWY_IS_UBSAN
return "ubsan";
#elif HWY_IS_DEBUG_BUILD
return "dbg";
#elif defined(__OPTIMIZE__) || defined(NDEBUG)
return "opt";
#else
return "unknown";
#endif
}

static inline void PinThreadToCore(size_t cpu_index) {
#if HWY_OS_LINUX
// Forces the thread to run on the logical processor with the same number.
Expand Down

0 comments on commit e760948

Please sign in to comment.