From f1bf916b8a00e07cbee4c2ff4634635f991185de Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Wed, 29 May 2024 23:12:35 -0400
Subject: [PATCH] apply NOOPT in test_arange complexity (#4774)

with hcopt, arange(2560) uses less ops than arange(256)
---
 test/test_arange.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/test_arange.py b/test/test_arange.py
index 4820fc0587..b9a12e44c9 100644
--- a/test/test_arange.py
+++ b/test/test_arange.py
@@ -1,10 +1,12 @@
 import unittest
 from tinygrad import Tensor, GlobalCounters
+from tinygrad.helpers import Context
 
 class TestArange(unittest.TestCase):
   def _get_flops(self, N):
     GlobalCounters.reset()
-    Tensor.arange(N).realize()
+    with Context(NOOPT=1):
+      Tensor.arange(N).realize()
     return GlobalCounters.global_ops
 
   def test_complexity(self):