add bigdl-llm-init (#9468)

This commit is contained in:
Lilac09 2023-11-15 14:55:33 +08:00 committed by GitHub
parent 0f82b8c3a0
commit 24146d108f

View file

@ -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"