portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-06-17 23:37:06 +01:00
Code Releases Activity

Bland SYCL compile

Browse Source
This commit is contained in:
Peter Boyle
2023-09-26 13:20:27 -07:00
parent 3d437c5cc4
commit 80359e0d49
5 changed files with 102 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
systems/OEM/benchmarks/select_gpu.sh Executable file
View File

@ -0,0 +1,13 @@
#!/bin/bash
num_tile=2
gpu_id=$(( (MPI_LOCAL_RANKID % num_tile ) ))
tile_id=$((MPI_LOCAL_RANKID / num_tile))
export ZE_AFFINITY_MASK=$gpu_id.$tile_id
echo "local rank $MPI_LOCALRANKID ; ZE_AFFINITY_MASK=$ZE_AFFINITY_MASK"
"$@"