happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use hwy::ThreadPool::MaxThreads() to determine the number of threads to use. 

PiperOrigin-RevId: 646117298
This commit is contained in:
The gemma.cpp Authors 2024-06-24 09:15:05 -07:00 committed by Copybara-Service
parent 12089417b5
commit ef786f1bfc
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
util/app.h
View File

@ -133,9 +133,8 @@ class AppArgs : public ArgsBase<AppArgs> {
}
static inline size_t GetSupportedThreadCount() {
return static_cast<size_t>(
std::clamp(static_cast<int>(std::thread::hardware_concurrency()) - 2, 1,
HWY_MIN(static_cast<int>(kMaxThreads), 18)));
return std::clamp(hwy::ThreadPool::MaxThreads(), size_t{1},
std::min(kMaxThreads, size_t{18}));
}
Path log; // output