1
0
Fork 0
Commit graph

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