Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
benchmark_mi250.sh 311 Bytes
Newer Older
chenxj's avatar
initial commit  
chenxj committed
1
2
3
4
5
6
7
8
9
10
11 
#!/bin/bash

#profile
HIP_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python3 benchmark_ait.py

#1GCD
HIP_VISIBLE_DEVICES=0 python3 benchmark_ait.py --batch-size "$1"

#2GCD
HIP_VISIBLE_DEVICES=0 python3 benchmark_ait.py --batch-size "$1" &
HIP_VISIBLE_DEVICES=1 python3 benchmark_ait.py --batch-size "$1" && fg