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