tinygrad/extra/optimization
chenyu 765f8b05e5
TernaryOps.WHERE has vin[0] as bool and BinaryOps.CMPLT always outputs bool (#2782)
* vin[0] to where is always bool

* due to better hack

* update test

* fix test_uops
2023-12-15 14:51:51 -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 TernaryOps.WHERE has vin[0] as bool and BinaryOps.CMPLT always outputs bool (#2782) 2023-12-15 14:51:51 -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