From 5cc536bcc0235a3a90ee9e57ec7b137fd113fd19 Mon Sep 17 00:00:00 2001
From: George Hotz <george@comma.ai>
Date: Mon, 30 Oct 2023 13:53:42 -0700
Subject: [PATCH] don't use locals applies to LASTLOCAL

---
 tinygrad/codegen/optimizer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/codegen/optimizer.py b/tinygrad/codegen/optimizer.py
index 4637f23bca..77a5b267a1 100644
--- a/tinygrad/codegen/optimizer.py
+++ b/tinygrad/codegen/optimizer.py
@@ -232,7 +232,7 @@ class OptimizedKernel(Kernel):
     amt = opt.amt if opt.amt != 0 else self.full_shape[axis]
     assert self.full_shape[axis] % amt == 0, "no longer valid shift"
     assert isinstance(amt, int) and amt != 1, "shift of amt 1 or Node is meaningless"
-    assert not self.dont_use_locals or opt.op not in {OptOps.LOCAL, OptOps.GROUPTOP, OptOps.UPCASTMID}, "not using locals"
+    assert not self.dont_use_locals or opt.op not in {OptOps.LOCAL, OptOps.LASTLOCAL, OptOps.GROUPTOP, OptOps.UPCASTMID}, "not using locals"
     if opt.op == OptOps.LOCAL:        # cyan
       assert axis < self.first_reduce, "can't local a reduce"
       assert not(self.tensor_core), "can't local with tensor cores"