Commit Graph

10417 Commits

Author SHA1 Message Date
George Hotz
ac8afd24fa refactor accel 2021-10-30 16:10:59 -07:00
George Hotz
082089d1c7 install requires pillow 2021-10-30 16:00:33 -07:00
Sebastian Kreft
8113eec4cf feat: add efficientnet test (#285)
Simple test using the Chicken example from https://upload.wikimedia.org/wikipedia/commons/4/41/Chicken.jpg and the image preprocessing from example/efficientnet.py

Note that EfficientNet loads the weights from the internet so running the tests may be slow the first time. We could speed up the tests by caching the /tmp folder.

Fixes #234
2021-10-30 15:53:51 -07:00
George Hotz
d8e5758c89 needs pillow 2021-10-27 10:14:02 -07:00
Sebastian Kreft
3358770182 chore(efficientnet): don't use eval when loading weights (#286)
Because the weights are being loaded from a third party internet address, it's unsafe to use eval. Also with the change I think the code became a little bit more clear as now it's clearer which keys are being transformed.

Co-authored-by: Seba Kreft <sebastian.kreft@houm.com>
2021-10-22 15:10:04 -07:00
Wenjun.Zhang
5402fd7f05 Fix: (#279)
1. python create model not match swift model in ane build folder.
2. fix ane lib path and add a dummy backward function for use_ane.py
can run in examples folder.
2021-08-06 07:58:13 -07:00
Zoli Veres
9f732e697a Fix loading relu.hwx from the wrong path caused by moving the ane folder into the accel folder (#276)
This is the commit that moved the ane folder 0eda3eb421
2021-06-29 07:20:54 -07:00
Guglielmo Camporese
2b7589db64 Added ResNet-{18, 34, 50, 101, 152} (#271)
* added resnets

* fix minor

* fix minor

* resnet in models

* added resnet test

* added resnet train test

* added linear, conv2d nn tests

* fix minor in extra/training

* resnet in models

* fix minor

* fix tolerance for linear in nn test

* fix eval, this causes cpu and gpu UT failing

* revert transformer test

* fix minor for CPU test

* improved model get_params for sequential layer

* fix minor for params counting

* commented broken ops tests

* improved train for resnet
2021-06-21 09:37:24 -07:00
George Hotz
89798d2f43 some flags 2021-06-19 11:46:31 -07:00
George Hotz
d81eae8288 debug cherry crash 2021-06-19 11:41:20 -07:00
George Hotz
0eda3eb421 move ane and fpga to accel folder 2021-06-19 11:26:20 -07:00
George Hotz
d3f169b267 move good models to models, add a training step test 2021-06-19 11:24:15 -07:00
George Hotz
b48d4bad2e clean up print spam 2021-06-19 10:31:04 -07:00
Jacky Lee
3a91d5434f Add dropout test (#265)
* Add dropout test

* Remove condition where training is false

* Skip dropout test when on GPU

* Revert changes to tensor.py and fix test case

* Revert change on whitespace

* Convert Tensor to cpu for testing

* Fix whitespace in tensor.py
2021-06-19 08:49:13 -07:00
George Hotz
ca0a38f2d5 more tpu notes 2021-06-18 13:28:06 -07:00
George Hotz
027535d0b5 microcoded matmul 2021-06-17 21:03:08 -07:00
George Hotz
d1dd9b46f6 if i was giving systolic arrays thumbs like siskyl and ebert (RIP) i would give them 0 2021-06-17 19:48:58 -07:00
George Hotz
026e2ae6a7 three registers and a zero command 2021-06-17 17:09:18 -07:00
George Hotz
2e71ae33f6 max op works 2021-06-17 17:01:21 -07:00
George Hotz
9e12c1bbba cherry binop 2021-06-17 16:50:40 -07:00
George Hotz
fcdabea880 training mnist with cherry ops 2021-06-17 16:45:35 -07:00
George Hotz
2affd226b3 speed up sum 2021-06-17 16:38:34 -07:00
George Hotz
e8eb7d1b7e max op 2021-06-17 16:20:56 -07:00
George Hotz
c1d469d440 sum op 2021-06-17 16:19:35 -07:00
George Hotz
d6517a8a7c ins 2021-06-16 19:31:13 -07:00
George Hotz
29a08ba352 pytorch earlier 2021-06-16 12:24:21 -07:00
George Hotz
4a07b71731 update business model 2021-06-16 12:01:50 -07:00
George Hotz
d29b16e5b4 more business notes 2021-06-16 11:47:57 -07:00
George Hotz
b1000d866e readme, plus reduce ops 2021-06-16 11:21:06 -07:00
George Hotz
ff3fdc58e5 risk -> cherry 2021-06-16 09:59:48 -07:00
George Hotz
2f91c012eb build note 2021-06-15 22:41:41 -07:00
George Hotz
0c02b66259 more 2021-06-15 15:02:32 -07:00
George Hotz
1e62e45d67 better todo 2021-06-15 10:30:16 -07:00
George Hotz
9ca4388695 debug 2021-06-15 10:24:21 -07:00
George Hotz
3d44aab52c more 2021-06-15 10:23:57 -07:00
George Hotz
4850d6eb43 update todo 2021-06-15 10:22:39 -07:00
George Hotz
4e1edb3692 have tinygrad log the loads 2021-06-14 18:35:14 -07:00
George Hotz
93f2e9769d little note 2021-06-14 15:49:41 -07:00
Jacky Lee
611d81dcb4 Add asserts for non-zero indices (#264) 2021-06-13 21:14:46 -07:00
George Hotz
508ced114c readme 2021-06-13 17:17:44 -07:00
Dinesh Kumar Gnanasekaran
2146860307 fixed OpenCL installation while running tests (#262)
Co-authored-by: dinesh <dinesh-GDK>
2021-06-12 11:14:21 -07:00
George Hotz
a89d12d735 wow, way faster 2021-06-10 17:11:39 -07:00
George Hotz
10b1306525 binops 2021-06-10 16:52:37 -07:00
George Hotz
4535d39baa comments and pow 2021-06-10 09:03:40 -07:00
George Hotz
2075fdeb4f FPGA Based Accelerator for Tinygrad (#258)
* ops_risk

* risk sim

* guessing is for winners

* minor

* better

* matmal with risk

* conv doesn't work

* closer

* conv2d works

* ops_risk

* opt2 works

* opt1 may not be possible

* opt1 is a mulacc

* arty

* attosoc example building on mac

* minor

* riscv assembler

* gucci gang

* we got C code

* not a scam

* hello

* make risk mergeable into master

* unop support
2021-06-07 17:45:09 -07:00
George Hotz
77ba198b57 Revert "Update README.md (#259)" (#260)
This reverts commit 5a69c5db6d.
2021-06-04 14:41:41 -07:00
Gabriel Rojas
5a69c5db6d Update README.md (#259) 2021-06-04 14:41:07 -07:00
Josh Smith
ad756f6112 minor optimizations & cleaning (#257)
* use isinstance, some optimizations & whitespace removal

* revert whitespace changes

* revert more whitespace

* some more cleanup

* revert fstring (not a fan of the {{}})

* fix typo

* fix typo
2021-06-02 09:57:15 -07:00
George Hotz
74e874cc0d comment 2021-05-26 18:06:55 -07:00
George Hotz
343c5f13c7 add output shape to DEBUG 2021-05-26 17:42:38 -07:00