github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-12 23:54:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

move renderer into options (#4514)

Browse Source 
* move renderer into options

* fix tests

* renders are functions
This commit is contained in:
George Hotz
2024-05-10 10:01:51 -07:00
committed by GitHub
parent 7c630a9a53
commit 4eef1ee9bf
15 changed files with 36 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/runtime/ops_gpu.py
View File

@@ -15,11 +15,10 @@ def check(status):
def checked(ret, status): return (check(status.value), ret)[1]
class CLCompiler(Compiler):
compiler_opts = CompilerOptions("GPU")
compiler_opts = CompilerOptions("GPU", renderer=OpenCLRenderer)
def __init__(self, device:CLDevice, compile_key:str):
self.device = device
super().__init__(f"compile_cl_{compile_key}")
def render(self, name:str, uops) -> str: return OpenCLRenderer(name, uops)
def compile(self, src:str) -> bytes:
program = checked(cl.clCreateProgramWithSource(self.device.context, 1, to_char_p_p([src.encode()]), None, status := ctypes.c_int32()), status)
build_status: int = cl.clBuildProgram(program, 1, self.device.device_id, None, cl.clBuildProgram.argtypes[4](), None)