mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-06-25 05:32:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

modeld_v2: add CUSTOM→CONST shape migration to pkl compat shim

Browse Source 
Old tinygrad used Ops.CUSTOM for scalar shape constants promoted to
vec types. New tinygrad uses Ops.CONST for the same representation.
This commit is contained in:
Jason Wen
2026-06-06 03:42:53 -04:00
parent b5daec54e2
commit 4ba620cdd2
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sunnypilot/modeld_v2/modeld.py
+2
View File
@@ -61,6 +61,8 @@ def _load_pkl_compat(data: bytes):
shape_uop = src[1]
if shape_uop.op is Ops.CONST and hasattr(shape_uop.dtype, 'count') and shape_uop.dtype.count > 1 and isinstance(shape_uop.arg, tuple):
src = (src[0], UOp(Ops.STACK, shape_uop.dtype, tuple(UOp(Ops.CONST, dtypes.weakint, (), arg=v) for v in shape_uop.arg)))
if op is Ops.CUSTOM and hasattr(dtype, 'count') and dtype.count > 1 and not isinstance(arg, str):
op = Ops.CONST
return _orig_call(cls, op, dtype, src, arg, tag, metadata, _buffer if op is Ops.BUFFER else None)
UOpMetaClass.__call__ = _compat_call