George Hotz e0201922e3 Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142) ... * stable diffusion < 324ms * revert swap action * fix tests due to more sum splitting * REDUCEOP_SPLIT_THRESHOLD env var * added from unaligned np test (#2134) * align cpu buffer before copy into cl buffer (#2135) * remove shelve from handcode_resnet50_opt.py (#2139) * Add dictionary keys to reduce db size (#2131) * work * ignore beam cache * dictionary keys are generic * minor db cleanups * fix baseline and extract dataset * fix training * log likelihood * more lin to feats * sts * training policynet * net sort of works * dedup * refactor, stupid new actions * fix uops deduping * BEAM_ESTIMATE --------- Co-authored-by: chenyu <chenyu@fastmail.com> Co-authored-by: imaolo <56898718+imaolo@users.noreply.github.com>		2023-10-27 10:53:06 -10:00
..
extract_policynet.py	Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)	2023-10-27 10:53:06 -10:00
extract_sa_pairs.py	Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)	2023-10-27 10:53:06 -10:00
generate_dataset.sh	op logger + replay (#2021)	2023-10-08 15:10:18 -07:00
get_action_space.py	optimization: get actions in CI (#2125)	2023-10-20 12:22:01 -07:00
helpers.py	Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)	2023-10-27 10:53:06 -10:00
pretrain_valuenet.py	Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)	2023-10-27 10:53:06 -10:00
rl.py	KOPT is over, BEAM is upstream (#2071)	2023-10-16 09:46:03 -07:00
run_qnet.py	Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)	2023-10-27 10:53:06 -10:00
test_beam_search.py	exclude ast with variables from beam search (#2140)	2023-10-25 16:35:29 -04:00
test_net.py	KOPT is over, BEAM is upstream (#2071)	2023-10-16 09:46:03 -07:00
test_time_linearizer.py	KOPT is over, BEAM is upstream (#2071)	2023-10-16 09:46:03 -07:00