256 cpus per task

Author: cquil11

runners/launch_mi325x-amd.sh

@@ -7,7 +7,7 @@ PARTITION="compute"
 SQUASH_FILE="/nfsdata/sa/squash/$(echo "$IMAGE" | sed 's/[\/:@#]/_/g').sqsh"
 
 set -x
-salloc --partition=$PARTITION --gres=gpu:$TP --time=180 --no-shell
+salloc --partition=$PARTITION --gres=gpu:$TP --cpus-per-task=256 --time=180 --no-shell
 JOB_ID=$(squeue -u $USER -h -o %A | head -n1)
 
 srun --jobid=$JOB_ID bash -c "sudo enroot import -o $SQUASH_FILE docker://$IMAGE"