github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-08 21:55:14 -05:00
Code Issues Packages Projects Releases Wiki Activity

oops, gpu not available

Browse Source
This commit is contained in:
George Hotz
2021-11-17 12:54:33 -08:00
parent e2a8961a18
commit 4320c45c4b
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/tensor.py
View File

@@ -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