commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hotfix: add TYPED=1 support

This commit is contained in:
George Hotz 2024-09-15 14:44:26 +08:00
parent 2d53e47b14
commit 5132bab48d
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tinygrad/__init__.py
View File

@ -1,6 +1,10 @@
from tinygrad.tensor import Tensor # noqa: F401
from tinygrad.engine.jit import TinyJit # noqa: F401
from tinygrad.shape.symbolic import Variable # noqa: F401
from tinygrad.dtype import dtypes # noqa: F401
from tinygrad.helpers import GlobalCounters, fetch, Context # noqa: F401
from tinygrad.device import Device # noqa: F401
import os
if int(os.getenv("TYPED", "0")):
from typeguard import install_import_hook
install_import_hook(__name__)
from tinygrad.tensor import Tensor # noqa: F401
from tinygrad.engine.jit import TinyJit # noqa: F401
from tinygrad.shape.symbolic import Variable # noqa: F401
from tinygrad.dtype import dtypes # noqa: F401
from tinygrad.helpers import GlobalCounters, fetch, Context, getenv # noqa: F401
from tinygrad.device import Device # noqa: F401