Commit Graph

1029 Commits

Author SHA1 Message Date
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 Hotz
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 Hotz
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
George Hotz
57e5df9f28 ane: procPath issue. don't waste more time with this, focus on core tinygrad 2022-08-16 10:36:13 +02:00
George Hotz
bdfdbc8f8d broken amfi patch 2022-08-13 10:41:25 +02:00
George Hotz
262efe5784 update readme 2022-08-09 11:08:52 +02:00
George Hotz
6267a3c8c2 notes 2022-08-09 00:42:14 +02:00
George Hotz
f4ff130947 docs 2022-08-09 00:06:24 +02:00
George Hotz
01de17eeb8 amfi note 2022-08-08 13:17:36 +02:00
George Hotz
136706169d fix ane on new mac os x 2022-08-06 19:10:22 +00:00
George Hotz
f300caa486 notes 2022-08-06 15:21:26 +00:00
George Hotz
27f209b80b add time sum 2022-08-06 14:35:21 +00:00
George Hotz
64c0d2dedb add gflop estimate 2022-08-06 14:21:48 +00:00
George Hotz
94d526f8fc fix op estimate 2022-08-06 14:15:50 +00:00
George Hotz
f2847cb710 remove useless init, add ops counter 2022-08-06 14:05:25 +00:00
George Hotz
18fde22dac fix that soon 2022-07-20 09:07:09 -07:00
George Hotz
187d581dc6 fix options on old pyopencl 2022-07-20 08:24:05 -07:00
Comma Device
6da956b9fa that should be right 2022-07-19 19:47:37 -07:00
Comma Device
f4ed837f2f float16 fixups 2022-07-19 19:44:40 -07:00
Comma Device
aa00a3948e needs_load in image correct 2022-07-19 19:25:47 -07:00
Comma Device
314d70ff17 zero out the buffer 2022-07-19 19:17:47 -07:00
Comma Device
b8a67905e5 save weights 2022-07-19 19:14:14 -07:00
George Hotz
46e7dfade1 REQUIRES_SIMPLE_REDUCE 2022-07-19 11:42:14 -07:00
George Hotz
acbeaf0ba9 adam in benchmark_train_efficientnet 2022-07-19 09:33:07 -07:00
George Hotz
ef1100fdff touchups 2022-07-19 09:30:06 -07:00
George Hotz
5d45c6e516 Fold reduce (#362)
* folding reduce

* fold through movementops

* fixup shapes

* was too aggressive

* i knew we needed that

* don't recompute reduce

* working

* fix openpilot compile

* prunegraph openpilot

* types and reduce_shape

* refactor

* cleanups

* neater

* 1009

* 1004

* clean up reduce for 998
2022-07-19 09:24:02 -07:00
Comma Device
2d402d1135 buffer_id is 8 bytes 2022-07-18 20:27:45 -07:00
Comma Device
577c23731e outputs with size 2022-07-18 20:21:33 -07:00
Comma Device
29581b5c85 inputs and outputs 2022-07-18 20:17:26 -07:00
Comma Device
ae30641b0d fix row pitch 2022-07-18 19:48:19 -07:00
Comma Device
02f23e526c output file to disk 2022-07-18 19:23:22 -07:00
George Hotz
ef4afdb5d2 tests maybe 2022-07-18 08:24:14 -07:00
George Hotz
a2c4bcf313 disable opencl tests 2022-07-18 08:17:21 -07:00
George Hotz
5093455166 don't shuffle if there's children involved 2022-07-17 21:15:40 -07:00
George Hotz
f76d41812b prune graph 2022-07-17 15:38:43 -07:00
George Hotz
eda6f071b2 default opt level 2 2022-07-17 14:54:40 -07:00
George Hotz
73b0471b25 join expands 2022-07-17 13:42:05 -07:00
George Hotz
cfabbbd6bb more crap to remove without convs 2022-07-17 13:02:27 -07:00
George Hotz
5e96ed523a fix opencl bug, no training on opencl 2022-07-17 12:55:26 -07:00
George Hotz
f93e297804 fix bug caused by rounding 2022-07-17 12:49:58 -07:00
George Hotz
cff297ef9d w/e, that's a later prob 2022-07-17 12:32:50 -07:00
George Hotz
4bc07326d4 we need that opt to make gpu decent speed 2022-07-17 12:26:18 -07:00
George Hotz
6375e7129a opencl not imported 2022-07-17 12:14:39 -07:00
George Hotz
bf299802f8 fixup tests 2022-07-17 12:11:53 -07:00
George Hotz
762e859089 testopencl 2022-07-17 11:56:40 -07:00
George Hotz
608e2431f7 test opencl, commit to removing the crap conv code from GPU 2022-07-17 11:55:37 -07:00
George Hotz
3c4565fa21 SLICE -> PAD,SHRINK 2022-07-17 11:33:59 -07:00
George Hotz
9574dd8559 some permutes are reshapes 2022-07-17 10:34:24 -07:00
George Hotz
77806e0d64 fix permute stacking 2022-07-17 10:24:57 -07:00