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

risk -> cherry

Browse Source
This commit is contained in:
George Hotz
2021-06-16 09:59:48 -07:00
parent 2f91c012eb
commit ff3fdc58e5
5 changed files with 59 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@@ -156,11 +156,11 @@ python3 -m pytest
### TODO (updated)
```bash
PYTHONPATH="." DEBUG=1 RISK=1 python3 examples/efficientnet.py https://upload.wikimedia.org/wikipedia/commons/4/41/Chicken.jpg
PYTHONPATH="." DEBUG=1 CHERRY=1 python3 examples/efficientnet.py https://upload.wikimedia.org/wikipedia/commons/4/41/Chicken.jpg
```
* Add reduce ops to RISK, and fully support forward pass. See `extra/ops_risk.py` and `extra/risk.py`
* Switch convolution backward pass to RISK instead of the numpy placeholder
* Confirm EfficientNet backward pass fully uses RISK instructions
* Add reduce ops to CHERRY, and fully support forward pass. See `extra/ops_risk.py` and `extra/risk.py`
* Switch convolution backward pass to CHERRY instead of the numpy placeholder
* Confirm EfficientNet backward pass fully uses CHERRY instructions
* Benchmark that and transformers