mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-06-01 07:25:49 +02:00
ggml-cpu : set openmp wait time if not set (llama/13758)
This commit is contained in:
parent
2d6c6862f7
commit
99596d6031
@ -3484,6 +3484,19 @@ void ggml_cpu_init(void) {
|
|||||||
const uint64_t t_end = ggml_time_us(); UNUSED(t_end);
|
const uint64_t t_end = ggml_time_us(); UNUSED(t_end);
|
||||||
|
|
||||||
GGML_PRINT_DEBUG("%s: GELU, Quick GELU, SILU and EXP tables initialized in %f ms\n", __func__, (t_end - t_start)/1000.0);
|
GGML_PRINT_DEBUG("%s: GELU, Quick GELU, SILU and EXP tables initialized in %f ms\n", __func__, (t_end - t_start)/1000.0);
|
||||||
|
|
||||||
|
#ifdef GGML_USE_OPENMP
|
||||||
|
//if (!getenv("OMP_WAIT_POLICY")) {
|
||||||
|
// // set the wait policy to active, so that OpenMP threads don't sleep
|
||||||
|
// putenv("OMP_WAIT_POLICY=active");
|
||||||
|
//}
|
||||||
|
|
||||||
|
if (!getenv("KMP_BLOCKTIME")) {
|
||||||
|
// set the time to wait before sleeping a thread
|
||||||
|
// this is less aggressive than setting the wait policy to active, but should achieve similar results in most cases
|
||||||
|
putenv("KMP_BLOCKTIME=200"); // 200ms
|
||||||
|
}
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
#if defined(__ARM_ARCH)
|
#if defined(__ARM_ARCH)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user