github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix row pitch

Browse Source
This commit is contained in:
Comma Device
2022-07-18 19:48:19 -07:00
parent 02f23e526c
commit ae30641b0d
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
openpilot/compile.py
View File

@@ -212,11 +212,12 @@ def compile(input, output_fn):
"id": ptr, "arg_type": "float*", "needs_load": False, "size": a.size,
})
elif isinstance(a, cl.Image):
#print(a.size, a.shape, a.row_pitch)
# TODO: modify thneed to respect no buffer_id here
# multiple of 32 isn't enough
row_pitch = (a.shape[0]*4*2 + 63)//64 * 64
size = row_pitch * a.shape[1]
jdat['objects'].append({
"id": ptr, "needs_load": False, "size": a.size, "arg_type": "image2d_t",
"width": a.shape[0], "height": a.shape[1], "row_pitch": a.row_pitch,
"id": ptr, "needs_load": False, "size": size, "arg_type": "image2d_t",
"width": a.shape[0], "height": a.shape[1], "row_pitch": row_pitch,
})
else:
raise Exception("unknown object", a)