mawei/openpilot
1
0
Fork 0
mirror of https://jihulab.com/mr-one/openpilot.git synced 2026-10-27 21:13:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

tinygrad: tolerate empty rewrite timings

Browse Source
This commit is contained in:
1okko
2026-10-07 11:50:13 +08:00
parent b0c74b1a93
commit 091fbfd351
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad_repo/tinygrad/engine/realize.py
+2 -1
View File
@@ -58,7 +58,8 @@ def estimate_uop(call:UOp) -> Estimates:
return Estimates()
first_run_cache:set[bytes] = set()
def track_stats(ctx:ExecContext, call:UOp, st:decimal.Decimal, ets:list[float|None]):
def track_stats(ctx:ExecContext, call:UOp, st:decimal.Decimal, ets:list[float|None]|None):
ets = ets or []
if ctx.update_stats:
is_hcq = (ast:=call.src[0]).op is Ops.CUSTOM_FUNCTION and ast.arg == "hcq"
estimates, n = estimate_uop(call), 1 if is_hcq else len(get_call_kernels(call))