Name Date Size #Lines LOC

..--

README.mdH A D25-Apr-2025809 1711

gen_data_mixed_mm.pyH A D25-Apr-20255 KiB147112

gen_mixedmm_heuristic_a100.shH A D25-Apr-2025113 62

gen_mixedmm_heuristic_h100.shH A D25-Apr-2025113 62

generate_heuristic_mixedmm.shH A D25-Apr-2025964 3115

get_mixedmm_dataset.shH A D25-Apr-2025344 1411

test_mixed_mm.pyH A D25-Apr-202516.5 KiB340308

train_decision_mixedmm.pyH A D25-Apr-20252.1 KiB5733

README.md

1If you just want to re-generate existing heuristics with already collected data for mixed_mm for A100/H100, run the following scripts:
2
3`bash get_mixedmm_dataset.sh # Downloads A100 and H100 datasets`
4`bash gen_mixedmm_heuristic_a100.sh # Generates A100 heuristic`
5`bash gen_mixedmm_heuristic_h100.sh # Generates H100 heuristic`
6
7If you want to collect new data, or generate a heuristic for another GPU, use the `generate_heuristic.sh` script:
8First, go into the generate_heuristic.sh and modify the variables according to the comments.
9Then run the script to perform benchmarks and collect training data:
10
11`bash generate_heuristic.sh collect`
12
13Depending on how many GPUs you are using, this might take a day.
14Afterwards, run the script in order to learn the heuristic:
15
16`bash generate_heuristic.sh generate`
17