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):