Commit Graph
10 Commits
Author SHA1 Message Date
George HotzandGitHub 791530045d Refactor LoadOps (#910)
* test

* work

* upd test

* loadops

* cleanups

* real ones

* remove LazyNumpyArray

* fix assign test

* remove range

* np.require

* llama uses arange kernels

* no caching consts

* fix enet

* torch load support

* tests cleanup

* fix shufflenet

* fix image

* fix torch_load test
2023-06-03 09:40:43 -07:00
George HotzandGitHub f5467cfedc Devicebufferless (#708)
* runs one metal kernel

* conv2d works

* ops tests are passing

* const folding

* all ops work

* pre commit always passes

* torch works

* working still

* fix graph test

* tests passing

* image almost works

* image conv works

* most images

* fix custom

* fix assignment

* fix compile enet

* clean up comments

* fix realize return value

* include shapetracker in LB repr

* copy should make a copy

* reenable method cache

* fix lna

* dtypes in graph

* forward only for IMAGE=2

* simple realize

* getting close

* fixup new api, it's good except the kernel count

* back to 197 kernels

* tests should pass

* go to a real float

* no type_on_cpu

* fix the docs

* put shapetracker back in it's proper place
2023-03-18 14:40:23 -07:00
geohot 3a8af99adb i understand ClassVar now 2023-03-15 09:00:25 -07:00
Pasan PereraandGitHub df48753692 fixed the import error for latest changes in master (#705) 2023-03-15 08:59:42 -07:00
George HotzandGitHub 54f499b623 Move rawbuffer (#697)
* move GlobalCounters to helpers

* that's not part of the public api

* move InterpretedBuffer

* remove fromCPU from devicebuffer
2023-03-13 22:30:36 -07:00
geohot cbc5a7222a symbolic is now a 6/10 due to the infinite loop. do better. 2023-03-13 00:07:59 -07:00
geohot c594a0a835 fix flip bug, add new unit tests 2023-03-12 23:55:31 -07:00
geohot ce1564b05e fix shapetracker test 2023-03-12 22:33:25 -07:00
geohot 153cce0f7e tutorial 2023-03-12 22:31:46 -07:00
geohot 8d16ebaea7 we have docs: 2023-03-12 19:05:44 -07:00