nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity
13,471 Commits 334 Branches 15 Tags
python_prg
Commit Graph
13471 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
geohot 114728d363 torch bs 2022-09-03 11:57:23 -07:00
geohot 356732515b stable_diffusion: add attn and layernorm 2022-09-03 11:02:27 -07:00
geohot 4dadd95e3c fix tests hopefully, more stable diffusion 2022-09-03 10:38:31 -07:00
geohot c01a8c5c2d stable diffusion start 2022-09-03 10:08:42 -07:00
Comma Device c07bf72d6e save free 200ms 2022-08-31 20:31:42 -04:00
George Hotz 2e9b7637b3 don't save input buffers 2022-08-31 15:37:38 -07:00
George Hotz a3fc64a585 fix batchnorm folding in openpilot compile 2022-08-31 13:04:49 -07:00
Comma Device a734df98fa TEST_ENET for openpilot compiler 2022-08-31 13:23:36 -04:00
George Hotz d919ac32af fix wrong size input 2022-08-31 09:07:34 -07:00
George Hotz 040640a580 fix cl import error 2022-08-31 08:43:44 -07:00
George Hotz e194ae0c1d typos 2022-08-30 19:52:21 -07:00
Mitchell GoffandGitHub 3af650b028 Rewrote Tensor.cat to be shorter and (hopefully) clearer (#372) 
* Rewrote Tensor.cat to be shorter and (hopefully) clearer

* Use cumsum[-1] instead of separate sum
 2022-08-30 16:15:07 -07:00
George Hotz db56297011 line count 2022-08-30 15:23:35 -07:00
George Hotz 33ac355bcd still broken 2022-08-29 19:08:07 -07:00
George Hotz 5efab7cf1d add reciprocal 2022-08-29 18:00:24 -07:00
George Hotz 880707f2d2 no torch test if no torch 2022-08-29 15:29:19 -07:00
George Hotz 5eba228844 print inputs 2022-08-29 08:56:04 -07:00
George Hotz fa33ba716e flip that 2022-08-28 11:27:33 -07:00
George Hotz dd587d26e3 oops, compare with abs 2022-08-28 11:23:21 -07:00
George Hotz dc7af8c3ac thneed run float32 2022-08-28 11:03:35 -07:00
Comma Device f0d11f29c7 float32 in image desc 2022-08-28 08:47:43 -07:00
Comma Device 9678cb8a1a hmm, the native exp/log breaks it too much 2022-08-22 17:13:08 -07:00
Comma Device 0c1378e7db native_exp is way faster on qcom 2022-08-22 16:58:50 -07:00
George Hotz 2162cd3383 fix typing 2022-08-22 16:25:15 -07:00
Comma Device e0a8d0f836 image input works 2022-08-22 16:04:17 -07:00
George Hotz 18340e7d30 remove from_image 2022-08-22 15:52:26 -07:00
Comma Device 1b5f4e52d9 refactor getters 2022-08-22 13:29:08 -07:00
Comma Device 13646ae07a opencl can't optimize that 2022-08-22 10:26:42 -07:00
George Hotz 227937d6d2 fix test maybe 2022-08-22 09:50:14 -07:00
George Hotz 11626053b0 run_thneed with test 2022-08-22 09:45:46 -07:00
kposborne2GitHubKyle
 		ec5d9b355c use functools.partialmethod (#369) 
Co-authored-by: Kyle <[email protected]>
 2022-08-21 12:13:31 -07:00
George Hotz e7a4cd91ba fix cpu thneed running 2022-08-21 12:11:07 -07:00
George Hotz 0aca848d10 enable the openpilot test 2022-08-21 12:05:48 -07:00
George Hotz a8734df030 add openpilot tests to tinygrad 2022-08-21 12:03:37 -07:00
geohot 7f15779942 t.assign in optim 2022-08-20 14:04:33 -07:00
geohot 5c8466f475 tune inter_red 2022-08-20 12:50:26 -07:00
geohot c5433db050 2 stage reduce 2022-08-20 12:42:11 -07:00
geohot b020fc2225 less needless reshaping 2022-08-20 08:43:38 -07:00
geohot ccd539e93e maybe that's a better way to do this 2022-08-20 08:25:12 -07:00
geohot b7d782b921 hmm, with the new reduce, we have to opt 3 for memory usage 2022-08-20 08:16:15 -07:00
geohot e34eb855fe much simpler reduce 2022-08-20 08:12:43 -07:00
geohot 1eb12dafbc reduce axis at the end 2022-08-20 07:40:56 -07:00
Comma Device 85453288d7 run_onnx_torch 2022-08-18 08:30:12 -07:00
Comma Device 1f23517d92 fixup run thneed 2022-08-18 08:22:53 -07:00
George HotzandGitHub b132de677d tinygrad.nn (#367) 
* tinygrad.nn

* flake8

* working on pylint

* more pylint

* more pylint

* pylint passes

* networkx

* mypy can't infer that type

* junk
 2022-08-18 07:41:00 -07:00
George HotzandGitHub 783c120a8c rawcpu (#365) 
* rawcpu

* add should work when we respect shapetracker

* now that's true

* still have to handle shapetracker

* copyin

* Fix mypy
 2022-08-17 11:33:20 +02:00
geohot 57e5df9f28 ane: procPath issue. don't waste more time with this, focus on core tinygrad 2022-08-16 10:36:13 +02:00
geohot bdfdbc8f8d broken amfi patch 2022-08-13 10:41:25 +02:00
geohot 262efe5784 update readme 2022-08-09 11:08:52 +02:00
geohot 6267a3c8c2 notes 2022-08-09 00:42:14 +02:00