From 4f6df37805a134db018e0ff63ee6437aa93b0ea1 Mon Sep 17 00:00:00 2001
From: Heyang Sun <60865256+Uxito-Ada@users.noreply.github.com>
Date: Wed, 3 Apr 2024 15:52:25 +0800
Subject: [PATCH] fix wrong cpu core num seen by docker (#10645)

---
 .../finetune/qlora/cpu/docker/start-qlora-finetuning-on-cpu.sh  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/llm/finetune/qlora/cpu/docker/start-qlora-finetuning-on-cpu.sh b/docker/llm/finetune/qlora/cpu/docker/start-qlora-finetuning-on-cpu.sh
index 90e0d885..697889b4 100644
--- a/docker/llm/finetune/qlora/cpu/docker/start-qlora-finetuning-on-cpu.sh
+++ b/docker/llm/finetune/qlora/cpu/docker/start-qlora-finetuning-on-cpu.sh
@@ -4,6 +4,7 @@ cd /ipex_llm
 export USE_XETLA=OFF
 export SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=1
 source /opt/intel/oneapi/setvars.sh
+export CPU_CORES=$(nproc)
 source ipex-llm-init -t
 
 if [ -d "./model" ];
@@ -19,7 +20,6 @@ fi
 if [ "$STANDALONE_DOCKER" = "TRUE" ]
 then
   export CONTAINER_IP=$(hostname -i)
-  export CPU_CORES=$(nproc)
   source /opt/intel/oneapi/setvars.sh
   export CCL_WORKER_COUNT=$WORKER_COUNT_DOCKER
   export CCL_WORKER_AFFINITY=auto