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

remove NumNode (#7035)

Browse Source
This commit is contained in:
chenyu
2024-10-13 16:42:19 -04:00
committed by GitHub
parent c4c806a210
commit bd8ecf7fd6
8 changed files with 17 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/external/fuzz_symbolic.py vendored
View File

@@ -1,8 +1,8 @@
import itertools
import random
from tinygrad import Variable
from tinygrad import Variable, dtypes
from tinygrad.ops import UOp
from tinygrad.helpers import DEBUG
from tinygrad.ops import NumNode
random.seed(42)
def add_v(expr, rng=None):
@@ -57,7 +57,7 @@ if __name__ == "__main__":
tape = [random.choice(ops) for _ in range(random.randint(2, 30))]
# 10% of the time, add one of lt, le, gt, ge
if random.random() < 0.1: tape.append(random.choice([lt, le, gt, ge]))
expr = NumNode(0)
expr = UOp.const(dtypes.int, 0)
rngs = []
for t in tape:
expr, rng = t(expr)