1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
#!/bin/bash
if [ $# -ne 1 ]; then
echo "Error: This script requires exactly one argument."
echo "`bash generate_heuristic_pad_mm.sh collect` to run benchmark and collect training data."
echo "`bash generate_heuristic_pad_mm.sh generate` to use the collected data to learn a heuristic."
exit 1
fi
MODE=$1
# !!! SPECIFY THE GPUs THAT YOU WANT TO USE HERE !!!
GPU_DEVICE_IDS="4,5"
# !!! SPECIFY THE CONDA ENVIRONEMNT THAT YOU WANT TO BE ACTIVATED HERE !!!
CONDA_ENV=heuristic-pr
NUM_SAMPLES=2000
# This is where AutoHeuristic will store autotuning results
OUTPUT_DIR="a100"
# !!! CHANGE THE NAME OF THE HEURISTIC IF YOU WANT TO LEARN A HEURISTIC FOR A GPU THAT IS NOT A100 !!!
HEURISTIC_NAME="PadMMA100"
BENCHMARK_SCRIPT="gen_data_pad_mm.py"
TRAIN_SCRIPT="train_regression_pad_mm.py"
bash ../generate_heuristic.sh ${MODE} ${GPU_DEVICE_IDS} ${CONDA_ENV} ${NUM_SAMPLES} ${OUTPUT_DIR} ${HEURISTIC_NAME} ${BENCHMARK_SCRIPT} ${TRAIN_SCRIPT}
|
