tinygrad/extra/optimization
George Hotz a0cb16ac61
node cleanup + local metal test speed [pr] (#6880)
* node cleanup [pr]

* fix tests, including the double one on metal

* no time tqdm tests
2024-10-04 18:14:23 +08:00
..
extract_dataset.py Revert "Revert "cleanup process_replay/* namings [run_process_replay] (#6429)…" (#6442) 2024-09-10 07:00:16 +08:00
extract_policynet.py lowerer is kernel [run_process_replay] (#5437) 2024-07-12 18:50:55 -07:00
extract_sa_pairs.py lowerer is kernel [run_process_replay] (#5437) 2024-07-12 18:50:55 -07:00
generate_dataset.sh Revert "Revert "cleanup process_replay/* namings [run_process_replay] (#6429)…" (#6442) 2024-09-10 07:00:16 +08:00
get_action_space.py remove iter from uopgraph (#6110) 2024-08-16 15:58:29 -07:00
helpers.py node cleanup + local metal test speed [pr] (#6880) 2024-10-04 18:14:23 +08:00
pretrain_valuenet.py lowerer is kernel [run_process_replay] (#5437) 2024-07-12 18:50:55 -07:00
rl.py s/get_linearizer/get_kernel [run_process_replay] (#5467) 2024-07-13 20:32:22 -04:00
run_qnet.py s/get_linearizer/get_kernel [run_process_replay] (#5467) 2024-07-13 20:32:22 -04:00
search.py update pylint path to check indent/space for all (#6022) 2024-08-10 14:41:09 -04:00
test_beam_search.py use float32 for rand buffer in test_beam_search and test in metal (#3831) 2024-03-19 23:22:58 -04:00
test_net.py s/get_linearizer/get_kernel [run_process_replay] (#5467) 2024-07-13 20:32:22 -04:00
test_time_linearizer.py s/get_linearizer/get_kernel [run_process_replay] (#5467) 2024-07-13 20:32:22 -04:00