diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index 0cb2496923..a6e5a12782 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -36,7 +36,10 @@ cl_ctx, cl_queue = None, None def require_init_gpu(): global cl_ctx, cl_queue if cl_queue is None: - cl_ctx = cl.create_some_context(interactive=False) + devices = cl.get_platforms()[0].get_devices(device_type=cl.device_type.GPU) + if len(devices) == 0: + devices = cl.get_platforms()[0].get_devices(device_type=cl.device_type.CPU) + cl_ctx = cl.Context(devices=devices) # this is an in-order command queue cl_queue = cl.CommandQueue(cl_ctx)