github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-13 08:05:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: else-return on runtime (#4881)

Browse Source 
* fix: add init file

* fix: no else-return

* fix: remove file again
This commit is contained in:
Roelof van Dijk
2024-06-08 14:44:24 +02:00
committed by GitHub
parent 1e3325f369
commit 1785a70e77
7 changed files with 14 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/ops_gpu.py
View File

@@ -66,7 +66,7 @@ class CLAllocator(LRUAllocator):
return checked(cl.clCreateImage2D(self.device.context, cl.CL_MEM_READ_WRITE,
cl.cl_image_format(cl.CL_RGBA, {2: cl.CL_HALF_FLOAT, 4: cl.CL_FLOAT}[options.image.itemsize]),
options.image.shape[1], options.image.shape[0], 0, None, status := ctypes.c_int32()), status)
else: return checked(cl.clCreateBuffer(self.device.context, cl.CL_MEM_READ_WRITE, size, None, status := ctypes.c_int32()), status)
return checked(cl.clCreateBuffer(self.device.context, cl.CL_MEM_READ_WRITE, size, None, status := ctypes.c_int32()), status)
def _free(self, buf:ctypes._CData, options:BufferOptions): check(cl.clReleaseMemObject(buf))
def copyin(self, dest:ctypes._CData, src:memoryview):
check(cl.clEnqueueWriteBuffer(self.device.queue, dest, False, 0, len(src)*src.itemsize, from_mv(src), 0, None, None))