From 4320c45c4bbca02bd2291f75c99d38343eb297ad Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 17 Nov 2021 12:54:33 -0800 Subject: [PATCH] oops, gpu not available --- tinygrad/tensor.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index 0bee35a538..075ee0ac4c 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -307,4 +307,7 @@ def _register_ops(namespace, device=Device.CPU): import importlib for d,ops in Device.imports.items(): - _register_ops(importlib.import_module('tinygrad.'+ops), d) + try: + _register_ops(importlib.import_module('tinygrad.'+ops), d) + except ImportError: + pass