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