github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Variable.num -> NumNode (#2354)

Browse Source
This commit is contained in:
chenyu
2023-11-18 15:45:52 -05:00
committed by GitHub
parent 40246d35bc
commit f02e17a967
8 changed files with 30 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/external/fuzz_symbolic.py vendored
View File

@@ -1,7 +1,7 @@
import itertools
import random
from tinygrad.helpers import DEBUG
from tinygrad.shape.symbolic import Variable
from tinygrad.shape.symbolic import Variable, NumNode
random.seed(42)
def add_v(expr, rng=None):
@@ -51,7 +51,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 = Variable.num(0)
expr = NumNode(0)
rngs = []
for t in tape:
expr, rng = t(expr)