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

define var (#3548)

Browse source 
* define var

* remove vars from there

* fix python symbolic ops

* fix llvm

* pypath
This commit is contained in:
George Hotz 2024-02-29 16:43:27 -08:00 • committed by GitHub
commit 2c19ab6561
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 26 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tinygrad/runtime/ops_python.py
View file

@ -56,6 +56,7 @@ class PythonProgram:
ul: Dict[int, Any] = {}
dl: Dict[int, DType] = {}
pbufs: List[memoryview] = list(bufs)
pvals: List[int] = list(vals)
i = 0
loop_ends: Dict[int, int] = {}
while i < len(self.uops):
@ -97,6 +98,8 @@ class PythonProgram:
assert dtype.fmt is not None
lbuf = memoryview(bytearray(arg[1]*dtype.itemsize))
ul[i] = [lbuf.cast(dtype.fmt)] * warp_size
elif uop is UOps.DEFINE_VAR:
ul[i] = [pvals.pop(0)] * warp_size
elif uop is UOps.SPECIAL:
if arg[1][0] == 'g':
ul[i] = [idxs[2-arg[0]]] * warp_size