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