From 7786ea9921170498c12b6dacec189dd2ee72df87 Mon Sep 17 00:00:00 2001 From: Peter Boyle Date: Tue, 3 Oct 2023 09:58:44 -0700 Subject: [PATCH] Bug fix in script --- systems/OEM/benchmarks/select_gpu.sh | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/systems/OEM/benchmarks/select_gpu.sh b/systems/OEM/benchmarks/select_gpu.sh index 2ef1f82d..db1cfc85 100755 --- a/systems/OEM/benchmarks/select_gpu.sh +++ b/systems/OEM/benchmarks/select_gpu.sh @@ -1,9 +1,8 @@ #!/bin/bash num_tile=2 - -gpu_id=$(( (MPI_LOCAL_RANKID % num_tile ) )) -tile_id=$((MPI_LOCAL_RANKID / num_tile)) +gpu_id=$(( (MPI_LOCALRANKID / num_tile ) )) +tile_id=$((MPI_LOCALRANKID % num_tile)) export ZE_AFFINITY_MASK=$gpu_id.$tile_id