node cleanup + local metal test speed [pr] (#6880)

* node cleanup [pr]

* fix tests, including the double one on metal

* no time tqdm tests
This commit is contained in:
George Hotz
2024-10-04 18:14:23 +08:00
committed by GitHub
parent cdff1d75b6
commit a0cb16ac61
10 changed files with 37 additions and 45 deletions

View File

@@ -41,7 +41,6 @@ def assert_same_lin(l1, l2):
# get features
import math
from tinygrad.shape.symbolic import Node
MAX_DIMS = 16
MAX_BUFS = 9
@@ -58,7 +57,7 @@ def lin_to_feats(lin:Kernel, use_sts=True):
# first, the full shape, including the colors
for s,os,c in zip(lin.full_shape,lin.output_shape,lc):
if isinstance(s, Node):
if isinstance(s, UOp):
ret.append(False)
ret += [0]*9
else: