Skip to content

GitLab

"megatron/data/bert_dataset.py" did not exist on "3e4e1ab2991bc5a395226d492f4eb6bb84be1256"
Switch branch/tag
Find file Normal viewHistoryPermalink
benchmark_mi250.sh 155 Bytes
Newer Older
chenxj's avatar
initial commit  
chenxj committed
1
2
3
4 
#!/bin/bash

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