Skip to content

Commit 144ae7c

Browse files
committed
binder
1 parent edfb0d4 commit 144ae7c

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

multi/binder.sh

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
#!/bin/bash
2+
case $(( ${OMPI_COMM_WORLD_LOCAL_RANK} )) in
3+
0) export UCX_NET_DEVICES=mlx5_0:1 ;;
4+
1) export UCX_NET_DEVICES=mlx5_1:1 ;;
5+
2) export UCX_NET_DEVICES=mlx5_2:1 ;;
6+
3) export UCX_NET_DEVICES=mlx5_3:1 ;;
7+
esac
8+
9+
echo Launching on $UCX_NET_DEVICES
10+
11+
$*
12+

0 commit comments

Comments
 (0)