Skip to content

GitLab

Commit cd1a6dfe authored by silencealiang's avatar silencealiang
Browse files

Update launch_with_binding.sh

parent e3cce568
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 6 deletions
requirements/launch_with_binding.sh
View file @ cd1a6dfe
#!/bin/bash #!/bin/bash
export HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
LOCAL_RANK=$1 LOCAL_RANK=$1
shift shift
gpu_map=(0 1 2 3 4 5 6 7)
numa_map=(0 1 2 3 4 5 6 7) numa_map=(0 1 2 3 4 5 6 7)
GPU_ID=${gpu_map[$LOCAL_RANK]}
NUMA_ID=${numa_map[$LOCAL_RANK]} NUMA_ID=${numa_map[$LOCAL_RANK]}
numactl --cpunodebind=${NUMA_ID} --membind=${NUMA_ID} "$@"
export HIP_VISIBLE_DEVICES=${GPU_ID}
numactl --cpunodebind=${NUMA_ID} --membind=${NUMA_ID} "$@"
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment