diff --git a/tinygrad/ops.py b/tinygrad/ops.py index 1e87926f0c..b9535bdb02 100644 --- a/tinygrad/ops.py +++ b/tinygrad/ops.py @@ -107,6 +107,9 @@ class Ops(FastEnum): EMPTY = auto() BUFFER_VIEW = auto() + # blocks in linearizer + BLOCK = auto(); BLOCKSTART = auto(); BLOCKFORK = auto(); BLOCKEND = auto() # noqa: E702 + EXPAND = auto() CONTRACT = auto() VIEW = auto()