chenyu a66ffec6d3 update kernel dataset to exclude the disktensor ones (#3651) ... disk tensor load contains big offset and is not meant to be run by gpu. repro steps ``` time ./extra/optimization/generate_dataset.sh gzip /tmp/sops mv /tmp/sops.gz extra/datasets/ ```		2024-03-07 17:35:19 -05: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	merge kernel and optimizer (#2200)	2023-11-01 15:20:01 -07:00
helpers.py	Revert "track size in shapetracker" (#3043)	2024-01-08 13:13:39 -08: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
test_beam_search.py	disable flaky test_conv_beam in CI (#3553)	2024-02-29 22:59:41 -05: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