Alnis Murtovi 7f1cda1533 Autoheuristic: Do not store choices as metadata (#130304) ...

While for optimizations like pad_mm, there are always only two possible choices, for other decision procedures, like kernel choice selection, the set of "available" choices depends on the input. Instead of storing the choices as metadata, we can instead take a look at all choices for which we have collected data (i.e. `df[CHOICE_COL].unique()`).

In this PR, I also try to replace "choice" and "feedback" with global constants CHOICE_COL and FEEDBACK_COL.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/130304
Approved by: https://github.com/eellison

2024-07-18 21:39:42 +00:00

..

gen_data_pad_mm.py

Autoheuristic: add config options for specifying optimizations to collect data for and use heuristics (#130245)

2024-07-18 01:04:36 +00:00

gen_pad_mm_a100.sh

Learn a heuristic to decide whether to pad before mm (#128643)

2024-07-15 23:04:06 +00:00

gen_pad_mm_h100.sh

Learn a heuristic to decide whether to pad before mm (#128643)

2024-07-15 23:04:06 +00:00

get_padmm_dataset.sh

Learn a heuristic to decide whether to pad before mm (#128643)

2024-07-15 23:04:06 +00:00

train_pad_mm.py

Learn a heuristic to decide whether to pad before mm (#128643)

2024-07-15 23:04:06 +00:00

train.py

Autoheuristic: Do not store choices as metadata (#130304)

2024-07-18 21:39:42 +00:00