Merge pull request #98 from leonardchan/tpu7x-auto

Add --numactl_bind flag to H2D benchmark script

Author: leonardchan

Ironwood/scripts/run_host_device_benchmark.sh

@@ -4,13 +4,15 @@
 CONFIG_DIR="Ironwood/configs/host_device"
 SPECIFIC_CONFIG=""
 INTERLEAVED=false
+NUMACTL_BIND=false
 
 # Helper function for usage
 usage() {
     echo "Usage: $0 [OPTIONS]"
     echo "Options:"
     echo "  --config <path>       Path to specific config file (optional)"
     echo "  --interleaved         Run with numactl --interleave=all"
+    echo "  --numactl_bind        Run with numactl --cpunodebind=0 --membind=0"
     echo "  --help                Show this help message"
     exit 1
 }
@@ -20,6 +22,7 @@ while [[ "$#" -gt 0 ]]; do
     case $1 in
         --config) SPECIFIC_CONFIG="$2"; shift ;;
         --interleaved) INTERLEAVED=true ;;
+        --numactl_bind) NUMACTL_BIND=true ;;
         --help) usage ;;
         *) echo "Unknown parameter passed: $1"; usage ;;
     esac
@@ -33,6 +36,7 @@ echo "********************************************************"
 echo ""
 echo "Configuration:"
 echo "    Interleaved: $INTERLEAVED"
+echo "    numactl Bound: $NUMACTL_BIND"
 echo ""
 
 if [ -n "$SPECIFIC_CONFIG" ]; then
@@ -44,6 +48,11 @@ else
     shopt -u nullglob
 fi
 
+if [[ "$INTERLEAVED" = true && "$NUMACTL_BIND" = true ]]; then
+    echo "Only one of --interleaved and --numactl_bind is allowed to be set at once."
+    exit 1
+fi
+
 if [ ${#CONFIGS[@]} -eq 0 ]; then
     echo "No configuration files found!"
     exit 1
@@ -61,6 +70,14 @@ for CONFIG_FILE in "${CONFIGS[@]}"; do
             echo "Warning: numactl not found. Running without interleaving."
             $CMD
         fi
+    elif [ "$NUMACTL_BIND" = true ]; then
+        if command -v numactl &> /dev/null; then
+            echo "Running with numactl --cpunodebind=0 --membind=0"
+            numactl --cpunodebind=0 --membind=0 $CMD
+        else
+            echo "Warning: numactl not found. Running without binding."
+            $CMD
+        fi
     else
         $CMD
     fi