mirror of
https://github.com/rui314/mold.git
synced 2024-09-21 18:08:01 +03:00
21 lines
462 B
C++
21 lines
462 B
C++
|
#include "mold.h"
|
||
|
|
|
||
|
|
#include <tbb/global_control.h>
|
||
|
|
|
||
|
|
namespace mold {
|
||
|
|
|
||
|
|
static i64 get_default_thread_count() {
|
||
|
|
// mold doesn't scale above 32 threads.
|
||
|
|
int n = tbb::global_control::active_value(
|
||
|
|
tbb::global_control::max_allowed_parallelism);
|
||
|
|
return std::min(n, 32);
|
||
|
|
}
|
||
|
|
|
||
|
|
void set_thread_count(i64 n) {
|
||
|
|
if (n == 0)
|
||
|
|
n = get_default_thread_count();
|
||
|
|
tbb::global_control tbb_cont(tbb::global_control::max_allowed_parallelism, n);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace mold
|