From fb4bd2a57d8c4eddacd10f1de16d44451b6c40ca Mon Sep 17 00:00:00 2001 From: chenyu Date: Fri, 19 Jan 2024 14:17:53 -0500 Subject: [PATCH] reenable padto to search action (#3183) --- tinygrad/features/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/features/search.py b/tinygrad/features/search.py index 50a066d9ba..e8a36f7d2c 100644 --- a/tinygrad/features/search.py +++ b/tinygrad/features/search.py @@ -14,7 +14,7 @@ actions = [Opt(op=OptOps.UPCAST, axis=axis, amt=amt) for amt in [0,2,3,4,7] for actions += [Opt(op=OptOps.UNROLL, axis=axis, amt=amt) for amt in [0,4] for axis in range(4)] actions += [Opt(op=OptOps.LOCAL, axis=axis, amt=amt) for amt in [2,3,4,8,13,16,29] for axis in range(5)] actions += [Opt(op=OptOps.GROUPTOP, axis=axis, amt=amt) for amt in [13,16,29,32,256] for axis in range(3)] -# actions += [Opt(op=OptOps.PADTO, axis=axis, amt=amt) for amt in [32] for axis in range(7)] +actions += [Opt(op=OptOps.PADTO, axis=axis, amt=amt) for amt in [32] for axis in range(7)] actions += [Opt(op=OptOps.LOCAL, axis=0, amt=32), Opt(op=OptOps.UPCASTMID, axis=1, amt=4), Opt(op=OptOps.GROUP, axis=0, amt=4), Opt(op=OptOps.GROUP, axis=0, amt=8), Opt(op=OptOps.GROUP, axis=1, amt=8),] if getenv("NOLOCALS"): actions += [Opt(op=OptOps.NOLOCALS)]