diff --git a/test/external/fuzz_symbolic.py b/test/external/fuzz_symbolic.py index 7c06fc1285..045f07e164 100644 --- a/test/external/fuzz_symbolic.py +++ b/test/external/fuzz_symbolic.py @@ -1,12 +1,12 @@ -import random, operator +import random, operator, sys import z3 from tinygrad import Variable, dtypes from tinygrad.uop.ops import UOp from tinygrad.uop.validate import uops_to_z3 from tinygrad.helpers import DEBUG, Context -seed = random.randint(0, 100) -print(f"Seed: {seed}") +seed = int(sys.argv[1]) if len(sys.argv) > 1 else random.randint(0, 100) +print(f"Seed: {seed}", flush=True) random.seed(seed) unary_ops = [lambda a:a+random.randint(-4, 4), lambda a: a*random.randint(-4, 4), diff --git a/test/external/fuzz_symbolic_symbolic_div.py b/test/external/fuzz_symbolic_symbolic_div.py index 693eb37e66..7a70ee34ce 100644 --- a/test/external/fuzz_symbolic_symbolic_div.py +++ b/test/external/fuzz_symbolic_symbolic_div.py @@ -1,11 +1,11 @@ -import random +import random, sys import z3 from tinygrad.uop.ops import UOp, Ops from tinygrad.uop.validate import uops_to_z3 from tinygrad.helpers import DEBUG, Context, colored -seed = random.randint(0, 100) -print(f"Seed: {seed}") +seed = int(sys.argv[1]) if len(sys.argv) > 1 else random.randint(0, 100) +print(f"Seed: {seed}", flush=True) random.seed(seed) def get_random_term(ranges, factors):