Grid/systems/PVC-OEM/benchmarks/select_gpu.sh

#!/bin/bash

num_tile=2
gpu_id=$(( (MPI_LOCALRANKID / num_tile ) ))
tile_id=$((MPI_LOCALRANKID % num_tile))

export ZE_AFFINITY_MASK=$gpu_id.$tile_id

echo "local rank $MPI_LOCALRANKID ; ZE_AFFINITY_MASK=$ZE_AFFINITY_MASK"

"$@"
Bland SYCL compile 2023-09-26 21:20:27 +01:00			`#!/bin/bash`

			`num_tile=2`
Bug fix in script 2023-10-03 17:58:44 +01:00			`gpu_id=$(( (MPI_LOCALRANKID / num_tile ) ))`
			`tile_id=$((MPI_LOCALRANKID % num_tile))`
Bland SYCL compile 2023-09-26 21:20:27 +01:00
			`export ZE_AFFINITY_MASK=$gpu_id.$tile_id`

			`echo "local rank $MPI_LOCALRANKID ; ZE_AFFINITY_MASK=$ZE_AFFINITY_MASK"`

			`"$@"`