This commit is contained in:
George Hotz
2021-06-17 16:20:56 -07:00
parent c1d469d440
commit e8eb7d1b7e
2 changed files with 2 additions and 1 deletions

View File

@@ -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 = {}