Commit Graph
19 Commits
Author SHA1 Message Date
George HotzandGitHub 902906f909 Fix constant folding (#713)
* fix

* codegen

* contiguous is real

* no bufs_to_delete

* don't assign rawconst

* remove neg and not

* need exec to fix custom function jit
2023-03-18 17:52:46 -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 37cf6fc4c0 err, external_test_opt.py broke...fusing will have to wait. correctness over speed 2023-03-11 17:54:47 -08:00
geohot 61071f881a fix bug, and add unit test to catch failure 2023-03-11 16:57:25 -08:00
geohot 3ec457248c failing llama test 2023-03-11 16:28:10 -08:00
geohot 01f39b19dc move to shapetracker.py 2023-03-11 07:50:07 -08:00
geohot d7cb8e3e56 multithreaded fake_torch_load_zipped 2023-03-10 19:16:27 -08:00
George Hotz 00641aa45d add challenge tests 2023-03-07 19:39:04 -08:00
George Hotz 4eb880550f enable contract test 2023-03-07 17:32:28 -08:00
George HotzandGitHub b561256a0e allow all reduces (#661)
* allow all reduces

* push permute tests

* explict permute reshape push

* contractw1s
2023-03-07 15:36:01 -08:00
geohot 7dbcc26582 fix up external tests 2023-03-06 06:52:28 -08:00
Alex WangandGitHub 64ecbd91b5 Refactor contraction and add integration test cases for push permute (#650)
* Refactor contraction and add unit tests

* Fix typo; Fix TestConv.test_elu failure due to some ones in old_shape

* Add push permute test cases

* Fix mypy type annotation check error

* Add contraction unit test; Reshape to higher dimension is not contraction
2023-03-06 06:36:55 -08:00
geohot b1ba78ac38 move applegpu disassembler 2023-03-05 11:21:12 -08:00
George HotzandGitHub 16b03f3c3b wow, can't believe that was broken (#642)
* wow, can't believe that was broken

* remove namedtuple comment
2023-03-04 22:28:28 -08:00
geohot 4a607f7d65 more ext gpu tests 2023-03-04 21:00:08 -08:00
geohot 69198a73d2 test_1x1_24_6 2023-03-04 20:37:46 -08:00
George HotzandGitHub b02a392d69 Improve local (#635)
* local is improving

* local is finding bugs

* new local should work
2023-03-04 09:30:49 -08:00
geohot 528cb3b3b9 fix ast test 2023-03-04 07:49:25 -08:00
geohot 8919ca8163 test cleanups 2023-03-03 06:36:06 -08:00