George Hotz 7103b716c4 merge kernel and optimizer (#2200) ...

* merge kernel and optimizer

* linearize is reentrant

* move global/local size

* clean up linearizer copy

* remove unneeded lin copies

* stop linearizing twice

* oops, that should be None

2023-11-01 15:20:01 -07:00

..

extract_policynet.py

merge kernel and optimizer (#2200)

2023-11-01 15:20:01 -07:00

extract_sa_pairs.py

merge kernel and optimizer (#2200)

2023-11-01 15:20:01 -07:00

generate_dataset.sh

op logger + replay (#2021)

2023-10-08 15:10:18 -07:00

get_action_space.py

merge kernel and optimizer (#2200)

2023-11-01 15:20: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

merge kernel and optimizer (#2200)

2023-11-01 15:20:01 -07: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

merge kernel and optimizer (#2200)

2023-11-01 15:20:01 -07: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