6 Commits

Author	SHA1	Message	Date
George Hotz	fff1f046b0	Simple version of the new GPU backend (#458) ... * newgpu * more to delete * hmm, tests pass with constant folding * fix lint/type * fix constant folding * comment and rerun tests * lazy touchups * fix graph_batchnorm test * smaller transformer to fix OOM * Revert "smaller transformer to fix OOM" This reverts commit a44ef8edc2. * no func cache * introspect * touchups * CLASTKernel * ugh, it was lru_cache * codegen * spacing * old gpu still in opencl * typing fix	2023-01-10 19:16:02 -08:00
George Hotz	7f15779942	t.assign in optim	2022-08-20 14:04:33 -07:00
George Hotz	1eb12dafbc	reduce axis at the end	2022-08-20 07:40:56 -07:00
George Hotz	f76d41812b	prune graph	2022-07-17 15:38:43 -07:00
George Hotz	f9a8412b68	make contiguous ops yellow	2022-07-02 17:54:04 -07:00
George Hotz	207b9e1df3	padding is now a param to conv2d	2022-07-02 17:11:12 -07:00