mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
max op
This commit is contained in:
@@ -38,7 +38,7 @@ from collections import defaultdict
|
||||
# <empty> <output> <input> <weight>
|
||||
# <weight> <input> <empty> <output>
|
||||
|
||||
SZ = 4
|
||||
SZ = 32
|
||||
SLOTSIZE = 1024*1024*2 # 5MB, for 20MB total. 8M elements
|
||||
sram = np.zeros((SLOTSIZE*4), dtype=np.float32)
|
||||
regfile = {}
|
||||
|
||||
Reference in New Issue
Block a user