Commit Graph
13471 Commits
Author SHA1 Message Date
geohot f193eeed25 bump all to python 3.9 2021-10-30 16:15:41 -07:00
geohot 63f50cff45 move back again 2021-10-30 16:13:29 -07:00
Evan MaysandGitHub 285621aeda Cherry backprop for conv2d (#281)
* quick math: 0 + x = x.

* gradient w.r.t. x using cherry for conv

* gradient w.r.t. w for conv on cherry but doing vector dot products

* small optimization

* [cherry] optimize conv backpass for large channel count

* get rid of numpy einsum
2021-10-30 16:12:19 -07:00
geohot 3d646272d6 move back 2021-10-30 16:12:12 -07:00
geohot ac8afd24fa refactor accel 2021-10-30 16:10:59 -07:00
geohot 082089d1c7 install requires pillow 2021-10-30 16:00:33 -07:00
Sebastian KreftandGitHub 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
geohot d8e5758c89 needs pillow 2021-10-27 10:14:02 -07:00
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 <[email protected]>
2021-10-22 15:10:04 -07:00
Wenjun.ZhangandGitHub 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 VeresandGitHub 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 https://github.com/geohot/tinygrad/commit/0eda3eb421458cae74b5ac75a24a9116ba192523
2021-06-29 07:20:54 -07:00
Guglielmo CamporeseandGitHub 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
geohot 89798d2f43 some flags 2021-06-19 11:46:31 -07:00
geohot d81eae8288 debug cherry crash 2021-06-19 11:41:20 -07:00
geohot 0eda3eb421 move ane and fpga to accel folder 2021-06-19 11:26:20 -07:00
geohot d3f169b267 move good models to models, add a training step test 2021-06-19 11:24:15 -07:00
geohot b48d4bad2e clean up print spam 2021-06-19 10:31:04 -07:00
Jacky LeeandGitHub 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
geohot ca0a38f2d5 more tpu notes 2021-06-18 13:28:06 -07:00
geohot 027535d0b5 microcoded matmul 2021-06-17 21:03:08 -07:00
geohot 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
geohot 026e2ae6a7 three registers and a zero command 2021-06-17 17:09:18 -07:00
geohot 2e71ae33f6 max op works 2021-06-17 17:01:21 -07:00
geohot 9e12c1bbba cherry binop 2021-06-17 16:50:40 -07:00
geohot fcdabea880 training mnist with cherry ops 2021-06-17 16:45:35 -07:00
geohot 2affd226b3 speed up sum 2021-06-17 16:38:34 -07:00
geohot e8eb7d1b7e max op 2021-06-17 16:20:56 -07:00
geohot c1d469d440 sum op 2021-06-17 16:19:35 -07:00
geohot d6517a8a7c ins 2021-06-16 19:31:13 -07:00
geohot 29a08ba352 pytorch earlier 2021-06-16 12:24:21 -07:00
geohot 4a07b71731 update business model 2021-06-16 12:01:50 -07:00
geohot d29b16e5b4 more business notes 2021-06-16 11:47:57 -07:00
geohot b1000d866e readme, plus reduce ops 2021-06-16 11:21:06 -07:00
geohot ff3fdc58e5 risk -> cherry 2021-06-16 09:59:48 -07:00
geohot 2f91c012eb build note 2021-06-15 22:41:41 -07:00
geohot 0c02b66259 more 2021-06-15 15:02:32 -07:00
geohot 1e62e45d67 better todo 2021-06-15 10:30:16 -07:00
geohot 9ca4388695 debug 2021-06-15 10:24:21 -07:00
geohot 3d44aab52c more 2021-06-15 10:23:57 -07:00
geohot 4850d6eb43 update todo 2021-06-15 10:22:39 -07:00
geohot 4e1edb3692 have tinygrad log the loads 2021-06-14 18:35:14 -07:00
geohot 93f2e9769d little note 2021-06-14 15:49:41 -07:00
Jacky LeeandGitHub 611d81dcb4 Add asserts for non-zero indices (#264) 2021-06-13 21:14:46 -07:00
geohot 508ced114c readme 2021-06-13 17:17:44 -07:00
2146860307 fixed OpenCL installation while running tests (#262)
Co-authored-by: dinesh <dinesh-GDK>
2021-06-12 11:14:21 -07:00
geohot a89d12d735 wow, way faster 2021-06-10 17:11:39 -07:00
geohot 10b1306525 binops 2021-06-10 16:52:37 -07:00
geohot 4535d39baa comments and pow 2021-06-10 09:03:40 -07:00
George HotzandGitHub 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 HotzandGitHub 77ba198b57 Revert "Update README.md (#259)" (#260)
This reverts commit 5a69c5db6d.
2021-06-04 14:41:41 -07:00