mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

apply NOOPT in test_arange complexity (#4774)

Browse source 
with hcopt, arange(2560) uses less ops than arange(256)
This commit is contained in:
chenyu 2024-05-29 23:12:35 -04:00 • committed by GitHub
commit f1bf916b8a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
test/test_arange.py
View file

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