File: acc_gpu_bind

package info (click to toggle)
mccode 3.5.19%2Bds5-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,113,256 kB
  • sloc: ansic: 40,697; python: 25,137; yacc: 8,438; sh: 5,405; javascript: 4,596; lex: 1,632; cpp: 742; perl: 296; lisp: 273; makefile: 226; fortran: 132
file content (13 lines) | stat: -rwxr-xr-x 305 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
 
#!/usr/bin/env bash
# Shellscript-hack to ensure mpi-distributed openacc processes
# get / initialize the right card (only)
#
# Kudos to Rémi Lacroix from the OpenACC_Hackathon slack 

if [ "x" == "x${CUDA_VISIBLE_DEVICES}" ];
then
    export CUDA_VISIBLE_DEVICES=${OMPI_COMM_WORLD_LOCAL_RANK}
fi

"$@"