Skip to content
Snippets Groups Projects
Commit 90984627 authored by ESCOBAR MUNOZ Juan's avatar ESCOBAR MUNOZ Juan
Browse files

Juan 17/11/2023:set_core_device/set_rocm_bind_device, Verbose only if BIND_VERBOSE not empty

parent 4d9bc889
No related branches found
No related tags found
1 merge request!4Jean Wurtz 30/04/2025 : Bugfixes mainly for TEB and for simple precision
...@@ -41,7 +41,7 @@ else ...@@ -41,7 +41,7 @@ else
export ACC_DEVICE_NUM=$(( LIP / SURBOOK )) export ACC_DEVICE_NUM=$(( LIP / SURBOOK ))
fi fi
echo IP=${IP} LIP=${LIP} NP=${NP} NN=${NN} NPN=${NPN} NPC=${NPC} HOST=`hostname` NB_CORE=${NB_CORE} CORE=${CORE} CORE_IMP=${CORE_IMP} HALF=$HALF SOC=$SOC RS=${RANK_SOC} ND=${NB_DEVICE} CD=${ACC_DEVICE_NUM} [[ "x${BIND_VERBOSE}" != "x" ]] && echo IP=${IP} LIP=${LIP} NP=${NP} NN=${NN} NPN=${NPN} NPC=${NPC} HOST=`hostname` NB_CORE=${NB_CORE} CORE=${CORE} CORE_IMP=${CORE_IMP} HALF=$HALF SOC=$SOC RS=${RANK_SOC} ND=${NB_DEVICE} CD=${ACC_DEVICE_NUM}
#EXEC=exec #EXEC=exec
${EXEC} ${Numactl} ${CORE} $* ${EXEC} ${Numactl} ${CORE} $*
......
...@@ -15,6 +15,6 @@ export NPN=$(( 1 + (NP-1)/ NN )) ...@@ -15,6 +15,6 @@ export NPN=$(( 1 + (NP-1)/ NN ))
export HN=$( hostname ) export HN=$( hostname )
export IG=$(( LIP / ( 1 + (NPN-1) / NB_DEVICE ) )) export IG=$(( LIP / ( 1 + (NPN-1) / NB_DEVICE ) ))
export ROCR_VISIBLE_DEVICES=${GPUSID[${IG}]} export ROCR_VISIBLE_DEVICES=${GPUSID[${IG}]}
echo LIP=${LIP} IP=${IP} NP=${NP} NN=${NN} NPN=${NPN} NG=${NB_DEVICE} IG=${IG} GPU=${ROCR_VISIBLE_DEVICES} ${HN} `taskset -pc $$` [[ "x${BIND_VERBOSE}" != "x" ]] && echo LIP=${LIP} IP=${IP} NP=${NP} NN=${NN} NPN=${NPN} NG=${NB_DEVICE} IG=${IG} GPU=${ROCR_VISIBLE_DEVICES} ${HN} `taskset -pc $$`
exec $* exec $*
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment