From 4bc07326d4ef839015f2e8133e63c7302d0af82f Mon Sep 17 00:00:00 2001 From: George Hotz Date: Sun, 17 Jul 2022 12:26:18 -0700 Subject: [PATCH] we need that opt to make gpu decent speed --- tinygrad/ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinygrad/ops.py b/tinygrad/ops.py index 25decfabb1..0b5ff00697 100644 --- a/tinygrad/ops.py +++ b/tinygrad/ops.py @@ -26,8 +26,8 @@ OPT = int(os.getenv("OPT", "1")) NOCONV = int(os.getenv("NOCONV", "0")) # TODO: movement ops that only change shape are really nops. treat them as such -REMOVE_MOVEMENT_NOPS, MERGE_UNARY_OPS = OPT>=1, OPT>=1 -MERGE_ELEMENTWISE_OPS, MERGE_ONE_CONV_INTO_ELEMENTWISE, MERGE_ELEMENTWISE_INTO_REDUCE, SHUFFLE_RESHAPE_OPS = OPT>=2, OPT>=2, OPT>=2, OPT>=2 +REMOVE_MOVEMENT_NOPS, MERGE_UNARY_OPS, MERGE_ELEMENTWISE_INTO_REDUCE = OPT>=1, OPT>=1, OPT>=1 +MERGE_ELEMENTWISE_OPS, MERGE_ONE_CONV_INTO_ELEMENTWISE, SHUFFLE_RESHAPE_OPS = OPT>=2, OPT>=2, OPT>=2 SHUFFLE_MOVEMENT_OPS = OPT>=3 SHUFFLE_PAD_OPS = OPT>=4 # NOTE: 0/0 is NaN if you pad, so this can change the output