github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

folding without UNMUL (#5628)

Browse Source 
* folding without UNMUL

* fix failures, index_collapse

* import ReduceOps

* test_arange_4096 isn't folding
This commit is contained in:
George Hotz
2024-07-21 20:14:44 -07:00
committed by GitHub
parent 583829ab44
commit 386fb5e7f8
6 changed files with 30 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_pattern_matcher.py
View File

@@ -1,7 +1,7 @@
import unittest, itertools
from test.helpers import TestUOps
from tinygrad.dtype import dtypes
from tinygrad.ops import BinaryOps, TernaryOps, UnaryOps # noqa: F401
from tinygrad.ops import BinaryOps, TernaryOps, ReduceOps, UnaryOps # noqa: F401
from tinygrad.codegen.uops import UOps, UOp, PatternMatcher, UPat, _match
from tinygrad.codegen.uopgraph import UOpGraph, constant_folder