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