commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

have tinygrad log the loads

This commit is contained in:
George Hotz 2021-06-14 18:35:14 -07:00
parent 93f2e9769d
commit 4e1edb3692
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
extra/risk.py
View File

@ -11,6 +11,7 @@
# 189 Mb embedded RAM, aka 9M 19-bit elements # 189 Mb embedded RAM, aka 9M 19-bit elements
# 2560 MLP blocks, 2 fp24 MULACC each # 2560 MLP blocks, 2 fp24 MULACC each
import os
import functools import functools
import numpy as np import numpy as np
from collections import defaultdict from collections import defaultdict
@ -161,9 +162,13 @@ def riski_matmul():
def riski_mov(tout, tin): def riski_mov(tout, tin):
regfile[tout][:] = regfile[tin] regfile[tout][:] = regfile[tin]
load_log = open("/tmp/risk_load_log", "w") if os.getenv("LOAD_LOG") else None
@count @count
def riski_load(target, address, stride_y=SZ, stride_x=1, len_y=SZ, len_x=SZ): def riski_load(target, address, stride_y=SZ, stride_x=1, len_y=SZ, len_x=SZ):
global util_n, util_d global util_n, util_d
if load_log is not None:
load_log.write("%d %d %d\n" % (address, stride_y, stride_x))
utils[(len_y, len_x)] += 1 utils[(len_y, len_x)] += 1
stride_y, stride_x = int(stride_y), int(stride_x) stride_y, stride_x = int(stride_y), int(stride_x)
d = regfile[target] d = regfile[target]