You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
12 lines
591 B
12 lines
591 B
4 days ago
|
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.uop.ops import UOp
|
||
|
|
Variable = UOp.variable
|
||
|
|
from tinygrad.dtype import dtypes # noqa: F401
|
||
|
|
from tinygrad.helpers import GlobalCounters, fetch, Context, getenv # noqa: F401
|
||
|
|
from tinygrad.device import Device # noqa: F401
|