chenyu 8798d120bb autopad shapetracker for BEAM (#2375) ...

* autopad shapetracker for BEAM

* OptOps.PADTO

* skip that test for now

* correct padding reduce axis

* just 32

* avoid more than double the FLOPs

* cleanups

* test case

* no support for triton and llvm yet

* typos

* symbolic shape would not work

* cannot PADTO with MAX kernel

* advance db version

* no breaking change - don't advance db version

* is triton just python?

* Revert "is triton just python?"

This reverts commit 17e776c25587615e33a3634c2fb0bb8591ce65d4.

* Revert "Revert "is triton just python?""

This reverts commit 6c434c01e1c4b0ea0431ec18632cd859fb3cf260.

* support llvm

* is it really passing in CI only?

* update tests

* oh triton test passed

* simpler

* revert that, with a test

* check if st are the same

* Revert "check if st are the same"

This reverts commit d2a5eac110a5da1af82a2728c883779ef69c3cad.

* update the db version

* rebase artifact

2023-11-22 21:05:25 -05: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

Simpler mops verify (#2325)

2023-11-15 21:47:18 -05: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

autopad shapetracker for BEAM (#2375)

2023-11-22 21:05:25 -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