23 строки
551 B
Bash
Executable File
23 строки
551 B
Bash
Executable File
#!/bin/bash
|
|
|
|
source /opt/hpcx/hpcx-init.sh
|
|
hpcx_load
|
|
|
|
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/AMD/amd-blis/lib/LP64/:/opt/hpcx/ompi/lib/:/opt/AMD/aocc-compiler-4.0.0/lib/
|
|
|
|
# On the off chance OMPI_MCA is set to UCX-only, disable that
|
|
unset OMPI_MCA_osc
|
|
|
|
BIN_PATH="$(dirname "$0")"
|
|
XHPL_EXE=$1
|
|
NCORES=$2
|
|
|
|
NT=${NCORES}
|
|
NR=2
|
|
MAP_BY=socket
|
|
set -x
|
|
|
|
mpirun --allow-run-as-root --map-by ${MAP_BY}:PE=$NT -np $NR --bind-to core \
|
|
-x OMP_NUM_THREADS=$NT -x OMP_PROC_BIND=close -x OMP_PLACES=cores \
|
|
${BIN_PATH}/bindmem.sh ${BIN_PATH}/${XHPL_EXE}
|