From 90984627d780fc99efd32bff6feb07bfedaa298a Mon Sep 17 00:00:00 2001
From: Juan ESCOBAR <juan.escobar@aero.obs-mip.fr>
Date: Fri, 17 Nov 2023 15:51:51 +0100
Subject: [PATCH] Juan 17/11/2023:set_core_device/set_rocm_bind_device, Verbose
only if BIND_VERBOSE not empty
---
bin/set_core_device | 2 +-
bin/set_rocm_bind_device | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/bin/set_core_device b/bin/set_core_device
index da1327453..b920ff01c 100755
--- a/bin/set_core_device
+++ b/bin/set_core_device
@@ -41,7 +41,7 @@ else
export ACC_DEVICE_NUM=$(( LIP / SURBOOK ))
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} ${Numactl} ${CORE} $*
diff --git a/bin/set_rocm_bind_device b/bin/set_rocm_bind_device
index cea4b811a..99d5cd38a 100755
--- a/bin/set_rocm_bind_device
+++ b/bin/set_rocm_bind_device
@@ -15,6 +15,6 @@ export NPN=$(( 1 + (NP-1)/ NN ))
export HN=$( hostname )
export IG=$(( LIP / ( 1 + (NPN-1) / NB_DEVICE ) ))
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 $*
--
GitLab