add bigdl-llm-init (#9468)
This commit is contained in:
parent
0f82b8c3a0
commit
24146d108f
1 changed files with 2 additions and 0 deletions
|
|
@ -171,6 +171,8 @@ else
|
|||
else
|
||||
# Logic for non-controller(worker) mode
|
||||
worker_address="http://$worker_host:$worker_port"
|
||||
# Apply optimizations from bigdl-llm
|
||||
source bigdl-llm-init -t
|
||||
# First check if user have set OMP_NUM_THREADS by themselves
|
||||
if [[ -n "${omp_num_threads}" ]]; then
|
||||
echo "Setting OMP_NUM_THREADS to its original value: $omp_num_threads"
|
||||
|
|
|
|||
Loading…
Reference in a new issue