From fa1c117c5bb5d73c6653336519af4f6f06d6e1c9 Mon Sep 17 00:00:00 2001
From: Pankaj Doharey <pankajdoharey@gmail.com>
Date: Sun, 26 Nov 2023 16:22:38 +0530
Subject: [PATCH 1/2] Update ops.py

Just assign the value for Enum directly. this should save so many auto() function calls.
---
 teenygrad/ops.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/teenygrad/ops.py b/teenygrad/ops.py
index e72c98a..d752bcc 100644
--- a/teenygrad/ops.py
+++ b/teenygrad/ops.py
@@ -1,12 +1,12 @@
 from enum import Enum, auto
 from typing import Optional
 
-class UnaryOps(Enum): NOOP = auto(); EXP2 = auto(); LOG2 = auto(); CAST = auto(); SIN = auto(); SQRT = auto(); RECIP = auto(); NEG = auto() # noqa: E702
-class BinaryOps(Enum): ADD = auto(); SUB = auto(); MUL = auto(); DIV = auto(); MAX = auto(); MOD = auto(); CMPLT = auto() # noqa: E702
-class ReduceOps(Enum): SUM = auto(); MAX = auto() # noqa: E702
-class TernaryOps(Enum): MULACC = auto(); WHERE = auto() # noqa: E702
-class MovementOps(Enum): RESHAPE = auto(); PERMUTE = auto(); EXPAND = auto(); PAD = auto(); SHRINK = auto(); STRIDE = auto() # noqa: E702
-class LoadOps(Enum): EMPTY = auto(); RAND = auto(); CONST = auto(); FROM = auto(); CONTIGUOUS = auto(); CUSTOM = auto() # noqa: E702
+class UnaryOps(Enum): NOOP, EXP2, LOG2, CAST, SIN, SQRT, RECIP, NEG = range(1, 8) # noqa: E702
+class BinaryOps(Enum): ADD, SUB, MUL, DIV, MAX, MOD, CMPLT = range(1, 7) # noqa: E702
+class ReduceOps(Enum): SUM, MAX = range(1, 2) # noqa: E702
+class TernaryOps(Enum): MULACC, WHERE = range(1, 2) # noqa: E702
+class MovementOps(Enum): RESHAPE, PERMUTE, EXPAND, PAD, SHRINK, STRIDE = range(1, 6) # noqa: E702
+class LoadOps(Enum): EMPTY, RAND, CONST, FROM, CONTIGUOUS, CUSTOM = range(1, 6) # noqa: E702
 
 class Device:
   DEFAULT = "CPU"

From f1d090457725d5b95104d88eb0f1f27c5fff7f27 Mon Sep 17 00:00:00 2001
From: Pankaj Doharey <pankajdoharey@gmail.com>
Date: Mon, 27 Nov 2023 12:41:38 +0530
Subject: [PATCH 2/2] Update ops.py

Updates range bound.
---
 teenygrad/ops.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/teenygrad/ops.py b/teenygrad/ops.py
index d752bcc..7211126 100644
--- a/teenygrad/ops.py
+++ b/teenygrad/ops.py
@@ -1,12 +1,12 @@
 from enum import Enum, auto
 from typing import Optional
 
-class UnaryOps(Enum): NOOP, EXP2, LOG2, CAST, SIN, SQRT, RECIP, NEG = range(1, 8) # noqa: E702
-class BinaryOps(Enum): ADD, SUB, MUL, DIV, MAX, MOD, CMPLT = range(1, 7) # noqa: E702
-class ReduceOps(Enum): SUM, MAX = range(1, 2) # noqa: E702
-class TernaryOps(Enum): MULACC, WHERE = range(1, 2) # noqa: E702
-class MovementOps(Enum): RESHAPE, PERMUTE, EXPAND, PAD, SHRINK, STRIDE = range(1, 6) # noqa: E702
-class LoadOps(Enum): EMPTY, RAND, CONST, FROM, CONTIGUOUS, CUSTOM = range(1, 6) # noqa: E702
+class UnaryOps(Enum): NOOP, EXP2, LOG2, CAST, SIN, SQRT, RECIP, NEG = range(1, 9) # noqa: E702
+class BinaryOps(Enum): ADD, SUB, MUL, DIV, MAX, MOD, CMPLT = range(1, 8) # noqa: E702
+class ReduceOps(Enum): SUM, MAX = range(1, 3) # noqa: E702
+class TernaryOps(Enum): MULACC, WHERE = range(1, 3) # noqa: E702
+class MovementOps(Enum): RESHAPE, PERMUTE, EXPAND, PAD, SHRINK, STRIDE = range(1, 7) # noqa: E702
+class LoadOps(Enum): EMPTY, RAND, CONST, FROM, CONTIGUOUS, CUSTOM = range(1, 7) # noqa: E702
 
 class Device:
   DEFAULT = "CPU"