github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

break out linearization [pr] (#6994)

Browse Source
This commit is contained in:
George Hotz
2024-10-11 15:27:33 +08:00
committed by GitHub
parent f319530191
commit e7a0ffe46a
10 changed files with 121 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_renderer_failures.py
View File

@@ -1,7 +1,8 @@
import unittest
from typing import List, cast
import numpy as np
from tinygrad.codegen.uopgraph import full_graph_rewrite, linearize_uop
from tinygrad.codegen.uopgraph import full_graph_rewrite
from tinygrad.codegen.linearize import linearize_uop
from tinygrad.device import Buffer, Device
from tinygrad.dtype import PtrDType, dtypes
from tinygrad.engine.realize import CompiledRunner