105 lines
4.0 KiB
Bash
105 lines
4.0 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Default
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 8 8 8 8 8 3 3
|
||
|
|
|
||
|
|
salloc -N 2 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 8 8 8 8 8 3 3
|
||
|
|
|
||
|
|
# Small N, C, H, W, K, R, S
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 1 1 8 8 1 4 4
|
||
|
|
|
||
|
|
salloc -N 2 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 1 1 8 8 1 4 4
|
||
|
|
|
||
|
|
# small N, K, large H, W, R, S
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 3 3 256 256 3 128 128
|
||
|
|
|
||
|
|
salloc -N 2 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 3 3 256 256 3 128 128
|
||
|
|
|
||
|
|
# small N, K, large H, W, R, S with padding, dilation and stride
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 2 -d 1 -s 1 3 3 256 256 3 128 128
|
||
|
|
|
||
|
|
salloc -N 2 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 5 -d 4 -s 1 3 3 256 256 3 128 128
|
||
|
|
|
||
|
|
# small H, W, R, S, large N, K
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 128 128 8 8 64 8 8
|
||
|
|
|
||
|
|
salloc -N 2 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 128 128 8 8 64 8 8
|
||
|
|
|
||
|
|
# small H, W, R, S, large N, K with padding, dilation and stride
|
||
|
|
salloc -N 2 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 4 -d 1 -s 2 128 128 8 8 64 8 8
|
||
|
|
|
||
|
|
salloc -N 2 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 10 -d 4 -s 2 128 128 8 8 64 8 8
|
||
|
|
|
||
|
|
# Odd N
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 7 4 64 64 4 16 16
|
||
|
|
|
||
|
|
# Odd C
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 4 9 64 64 4 16 16
|
||
|
|
|
||
|
|
# Odd H W
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 4 2 123 69 4 16 16
|
||
|
|
|
||
|
|
# Odd K
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 4 2 64 64 9 16 16
|
||
|
|
|
||
|
|
# Odd R, S
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 0 -d 1 -s 1 4 2 64 64 7 19 37
|
||
|
|
|
||
|
|
# Random
|
||
|
|
salloc -N 1 --exclusive \
|
||
|
|
mpirun --bind-to none -mca btl ^openib -npernode 1 \
|
||
|
|
numactl --physcpubind 0-39 \
|
||
|
|
./main -v -p 4 -d 2 -s 3 3 7 63 121 7 19 28
|
||
|
|
|