Francis Lam 6d5dec2fef log optimized kernels and a script to compare with non-optimized ones (#3829) ... * search: add BEAM_VERIFY option to validate search results refactor fuzz_linearizer comparison to allow it to be used in for BEAM_VERIFY in device.py * search: fix to verify the beam_search result and not the fastest * search: fix typing and clean up * device: remove imports from test and add LOGKERN options LOGKERN output can be used with test/external/verify_kernel.py to validate correctness * fix example in verify_kernel.py * cleanup fixes * fix to use f-strings		2024-03-20 19:22:08 -04:00
..
extract_policynet.py	move dtypes to dtype.py (#2964)	2024-01-01 14:58:48 -08:00
extract_sa_pairs.py	move dtypes to dtype.py (#2964)	2024-01-01 14:58:48 -08:00
generate_dataset.sh	update kernel dataset to exclude the disktensor ones (#3651)	2024-03-07 17:35:19 -05:00
get_action_space.py	multioutput uoping infra (#3706)	2024-03-15 21:56:59 -07:00
helpers.py	log optimized kernels and a script to compare with non-optimized ones (#3829)	2024-03-20 19:22:08 -04:00
pretrain_valuenet.py	move dtypes to dtype.py (#2964)	2024-01-01 14:58:48 -08:00
rl.py	fixed imports (#2185)	2023-10-30 22:07:17 -07:00
run_qnet.py	Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)	2023-10-27 10:53:06 -10:00
search.py	search: add a tool that beam searches one or more kernels (#3685)	2024-03-11 16:02:17 -04:00
test_beam_search.py	use float32 for rand buffer in test_beam_search and test in metal (#3831)	2024-03-19 23:22:58 -04:00
test_net.py	fixed imports (#2185)	2023-10-30 22:07:17 -07:00
test_time_linearizer.py	KOPT is over, BEAM is upstream (#2071)	2023-10-16 09:46:03 -07:00