github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

zero out the inputs

Browse Source
This commit is contained in:
George Hotz
2022-10-20 13:46:52 -07:00
parent 61ee428e4c
commit a18c1f3178
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extra/thneed.py
View File

@@ -111,10 +111,10 @@ class Thneed:
ptr = nptr
# populate the cl_cache
for k in jdat['kernels']:
for i,k in enumerate(jdat['kernels']):
kernel = prgs[k['name']]
aaa = []
for i,(a,sz) in enumerate(zip(k['args'], k['args_size'])):
for j,(a,sz) in enumerate(zip(k['args'], k['args_size'])):
if len(a) == 0:
aa = cl.LocalMemory(sz)
elif len(a) == 4:
@@ -124,6 +124,7 @@ class Thneed:
a = a.encode('latin_1')
aa = np.uint16(struct.unpack("H", a)[0])
elif len(a) == 8:
#print(i,j,struct.unpack("Q", a.encode('latin_1'))[0])
aa = bufs[a]
aaa.append(aa)
self.cl_cache.append((kernel, [k['global_work_size'], k['local_work_size'], *aaa]))