22 lines
436 B
Bash
22 lines
436 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
: ${NODES:=2}
|
||
|
|
|
||
|
|
salloc -N $NODES --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main $@
|
||
|
|
|
||
|
|
# N C K H W: graetaer than 32, power of 2
|
||
|
|
: ${N:=32}
|
||
|
|
: ${K:=64}
|
||
|
|
: ${C:=64}
|
||
|
|
: ${H:=256}
|
||
|
|
: ${W:=256}
|
||
|
|
|
||
|
|
# R S: 16
|
||
|
|
: ${R:=16}
|
||
|
|
: ${S:=16}
|
||
|
|
|
||
|
|
# order of parameters: N C H W K R S
|
||
|
|
source ./run.sh -d 1 -p 0 -s 1 -n 1 $N $C $H $W $K $R $S $@
|