mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

swap src[2] and src[3] in load [run_process_replay] (#5821)

Browse source 
* swap src[2] and src[3] in load [run_process_replay]

* cleanups + bugfix

* fix ptx
This commit is contained in:
George Hotz 2024-07-30 14:04:13 -07:00 • committed by GitHub
commit 693990a346
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 30 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tinygrad/runtime/ops_python.py
View file

@ -18,7 +18,7 @@ def _load(m, i):
return m[i]
def load(inp, j=0):
if len(inp) == 4: return [_load(m, x+j) if gate else default for m,x,gate,default in zip(*inp)]
if len(inp) == 4: return [_load(m, x+j) if gate else default for m,x,default,gate in zip(*inp)]
return [_load(m, x+j) for m,x in zip(inp[0], inp[1])]
def _store(m, i, v):