github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 14:43:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

30 MEGAReLUs. we need to lose 12 lines

Browse Source
This commit is contained in:
George Hotz
2020-12-12 17:07:34 -08:00
parent 49da969d25
commit a5aced8d47
9 changed files with 73 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -89,7 +89,7 @@ from tinygrad.tensor import Tensor
### ANE Support?!?!
If all you want to do is ReLU, you are in luck! You can do very fast ReLU (fastness not confirmed)
If all you want to do is ReLU, you are in luck! You can do very fast ReLU (at least 30 MEGAReLUs/sec confirmed)
Requires your Python to be signed with `ane/lib/sign_python.sh` to add the `com.apple.ane.iokit-user-access` entitlement.
@@ -101,6 +101,8 @@ b = a.relu()
print(b.cpu())
```
Warning: do not rely on the ANE port. It segfaults sometimes. So if you were doing something important with tinygrad and wanted to use the ANE, you might have a bad time.
### ImageNet inference
Despite being tiny, tinygrad supports the full EfficientNet. Pass in a picture to discover what it is.