UOp.variable (#7010)

* UOp.variable [pr]

* fix tests

* clean

* improve name rendering

* last bug
This commit is contained in:
George Hotz
2024-10-12 18:20:44 +08:00
committed by GitHub
parent f79e05cac0
commit 5ae2de9845
19 changed files with 52 additions and 49 deletions

View File

@@ -1,12 +1,13 @@
# stuff needed to unpack a kernel
from typing import Tuple
from extra.ops import LazyOp, TernaryOps, BinaryOps, UnaryOps, ReduceOps, BufferOps, MemBuffer, ConstBuffer, MetaOps
from tinygrad import Variable
from tinygrad.codegen.kernel import Opt, OptOps
from tinygrad.ops import UOp, UOps, KernelInfo
from tinygrad.dtype import dtypes, PtrDType
from tinygrad.shape.shapetracker import ShapeTracker
from tinygrad.shape.view import View
from tinygrad.shape.symbolic import Variable, NumNode
from tinygrad.shape.symbolic import NumNode
inf, nan = float('inf'), float('nan')
# kernel unpacker

View File

@@ -2,8 +2,7 @@ import unittest
import numpy as np
from tinygrad.helpers import BEAM, Timing, CI
from tinygrad.shape.symbolic import Variable
from tinygrad.tensor import Tensor
from tinygrad import Variable, Tensor
from tinygrad.nn import Conv2d
def rand(*shape):