7 Commits

Author	SHA1	Message	Date
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
George Hotz	abeba8f1fc	optimization: get actions in CI (#2125) ... * get actions in CI * actually run the test * pythonpath	2023-10-20 12:22:01 -07:00
George Hotz	c36d306606	KOPT is over, BEAM is upstream (#2071) ... * create cache for q learning * make linter happy * global beam * where it belongs * bugfix * ditch the kopt, use the beam * faster lin and DEBUG=2 okay * remove kopt, move search to features	2023-10-16 09:46:03 -07:00
George Hotz	49bcfec383	0s in the action space (#2070) ... * 0s in the action space * simpler * skip duplicate actions	2023-10-14 11:22:48 -07:00
George Hotz	4124cf1df5	cleanup tensor cores, expose exclude local upcast (#2064) ... * expose exclude_local_upcast * convert apply tensor cores to ops * update comment * put LOCAL back to what it was, BEAM is better than way	2023-10-14 09:21:03 -07:00
George Hotz	90c777d815	remove apply_auto_opt (#2063)	2023-10-13 07:44:14 -07:00
George Hotz	6f1810af2d	with unroll, the action space goes from 161 -> 127 (#2060) ... * with unroll, the action space goes from 161 -> 127 * more reliable instrumentation * beam search is so op * beam bugfix	2023-10-12 20:52:23 -07:00