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

CL_DEVICE option

Browse Source
This commit is contained in:
George Hotz
2022-06-24 23:22:10 -07:00
parent d748353ce5
commit 3a147137ee
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tinygrad/llops/ops_gpu.py
View File

@@ -12,10 +12,10 @@ class CL:
CACHE = None
def __init__(self):
if getattr(CL, "cl_queue", None) is not None: return
devices = cl.get_platforms()[0].get_devices(device_type=cl.device_type.GPU)
devices = sum([x.get_devices(device_type=cl.device_type.GPU) for x in cl.get_platforms()], [])
if len(devices) == 0: # settle for CPU
devices = cl.get_platforms()[0].get_devices(device_type=cl.device_type.CPU)
CL.cl_ctx = cl.Context(devices=devices)
devices = sum([x.get_devices(device_type=cl.device_type.CPU) for x in cl.get_platforms()], [])
CL.cl_ctx = cl.Context(devices=[devices[int(os.getenv("CL_DEVICE", "0"))]])
CL.cl_queue = cl.CommandQueue(self.cl_ctx) # this is an in-order command queue
@staticmethod