diff --git a/README.md b/README.md index 31ec64f2d2..98a27146e4 100644 --- a/README.md +++ b/README.md @@ -160,6 +160,7 @@ PYTHONPATH="." DEBUG=1 CHERRY=1 python3 examples/efficientnet.py https://upload. ``` * ~~Add reduce ops to CHERRY, and fully support forward pass. See `extra/ops_risk.py` and `extra/risk.py`~~ +* Fix max op * Switch convolution backward pass to CHERRY instead of the numpy placeholder * Confirm EfficientNet backward pass fully uses CHERRY instructions * Benchmark that and transformers diff --git a/extra/cherry.py b/extra/cherry.py index 61d88787ae..e6efc708ef 100755 --- a/extra/cherry.py +++ b/extra/cherry.py @@ -38,7 +38,7 @@ from collections import defaultdict # # -SZ = 4 +SZ = 32 SLOTSIZE = 1024*1024*2 # 5MB, for 20MB total. 8M elements sram = np.zeros((SLOTSIZE*4), dtype=np.float32) regfile = {}