github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-04 19:55:18 -05:00
Code Issues Packages Projects Releases Wiki Activity

remove iter from uopgraph (#6110)

Browse Source 
* remove iter from uopgraph

* linearize returns uops

* fix tests

* linearize in linearize

* tests fix

* touchup

* test failures
This commit is contained in:
George Hotz
2024-08-16 15:58:29 -07:00
committed by GitHub
parent 28c75bf2a6
commit 74ee9febec
15 changed files with 104 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_device_speed.py
View File

@@ -1,13 +1,12 @@
import unittest
from tinygrad import Device
from tinygrad.codegen.uopgraph import UOpGraph
from tinygrad.helpers import Timing, Profiling
class TestDeviceSpeed(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.dev = Device[Device.DEFAULT]
cls.empty = Device[Device.DEFAULT].renderer.render("test", UOpGraph([]))
cls.empty = Device[Device.DEFAULT].renderer.render("test", [])
def test_empty_compile(self):
with Timing("compiler "):