From f4faf401bc50573aed1a034ac8d12bf7cba7112f Mon Sep 17 00:00:00 2001 From: Skosh Date: Fri, 11 Dec 2020 19:21:59 +0200 Subject: [PATCH] require_init_gpu() function selects GPU as device and falls back to CPU if none are available (#180) * require_init_gpu() function selects GPU as device and falls back to CPU if none are available * Small fix for CPU specific code * Should work... --- tinygrad/tensor.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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)