xref: /aosp_15_r20/external/pytorch/torchgen/_autoheuristic/mm/get_mm_dataset.sh (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1#!/bin/bash
2
3base_url='https://github.com/AlnisM/autoheuristic-datasets/raw/main/'
4a100_data='a100_mm.zip'
5h100_data='h100_mm.zip'
6datasets=("${a100_data}" "${h100_data}")
7for dataset in "${datasets[@]}"; do
8    url="${base_url}${dataset}"
9    wget ${url}
10    unzip ${dataset}
11    rm ${dataset}
12done
13