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