geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

fix define global (#4383)

Browse Source 
* fix define global

* remove name from DEFINE_GLOBAL

* fix fuzzing

* fix ptx

* fix python
This commit is contained in:
George Hotz
2024-05-01 22:32:56 -04:00
committed by GitHub
parent ad116dc5c6
commit f635c4d273
10 changed files with 48 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_python.py
+1 -1
View File
@@ -71,7 +71,7 @@ class PythonProgram:
dl[i] = dtype
if uop is UOps.DEFINE_GLOBAL:
assert dtype.fmt is not None
ul[i] = [pbufs[arg[0]].cast(dtype.fmt)] * warp_size
ul[i] = [pbufs.pop(0).cast(dtype.fmt)] * warp_size
elif uop is UOps.DEFINE_LOCAL:
assert dtype.fmt is not None
lbuf = memoryview(bytearray(arg[1]*dtype.itemsize))