flip order, this is simpler

This commit is contained in:
2026-05-04 20:02:41 -07:00
parent cfefef479b
commit 437205ae03
7 changed files with 11 additions and 11 deletions
+1 -1
View File
@@ -18,7 +18,7 @@ def _load(m, i, dtype: DType):
return from_storage_scalar(m[i], dtype)
def load(inp, j, dtype: DType):
if len(inp) >= 3: return [_load(m, x+j if x is not None else None, dtype) if gate else default for (m,x),gate,default in zip(*inp[:3])]
if len(inp) >= 3: return [_load(m, x+j if x is not None else None, dtype) if gate else default for (m,x),default,gate in zip(*inp[:3])]
return [_load(m, x+j if x is not None else None, dtype) for m,x in inp[0]]
def _store(m, i, v, dtype: DType):