From 8a0fb3d765b60f739ed261cc286301c98419f66e Mon Sep 17 00:00:00 2001 From: chenyu Date: Sat, 11 May 2024 13:06:10 -0400 Subject: [PATCH] delete old extra/autopad.py (#4532) --- extra/autopad.py | 37 ------------------------------------- 1 file changed, 37 deletions(-) delete mode 100644 extra/autopad.py diff --git a/extra/autopad.py b/extra/autopad.py deleted file mode 100644 index 48a97f9231..0000000000 --- a/extra/autopad.py +++ /dev/null @@ -1,37 +0,0 @@ -from tinygrad.tensor import Tensor -from tinygrad.ops import LoadOps -from tinygrad.codegen.linearizer import Linearizer -from test.external.fuzz_linearizer import run_linearizer -from tinygrad.codegen.kernel import Opt, OptOps -from tinygrad.engine.schedule import create_schedule - -N = 17**3 - -a = Tensor.rand(N, N) -b = Tensor.rand(N, N) -c = a @ b -sched = [si for si in create_schedule([c.lazydata]) if si.ast.op not in LoadOps] -assert len(sched) == 1 -lin = Linearizer(sched[0].ast) - -lin.apply_opt(Opt(op=OptOps.PADTO, axis=0, amt=32)) -lin.apply_opt(Opt(op=OptOps.PADTO, axis=1, amt=32)) -lin.hand_coded_optimizations() -lin.linearize() -print(f"{lin.applied_opts=}") - -run_linearizer(lin) - -### - -a = Tensor.rand(61, 61).sum(axis=0) -sched = [si for si in create_schedule([a.lazydata]) if si.ast.op not in LoadOps] -assert len(sched) == 1 -lin = Linearizer(sched[0].ast) - -lin.apply_opt(Opt(op=OptOps.PADTO, axis=0, amt=32)) -lin.hand_coded_optimizations() -lin.linearize() -print(f"{lin.applied_opts=}") - -run_linearizer(lin) \ No newline at end of file