tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 15:36:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

mulacc_unrolled should happen even with no DEVECTORIZE (#9029)

Browse Source 
* mulacc_unrolled should happen even with no DEVECTORIZE

* Update rewriter.py

* Update rewriter.py

---------

Co-authored-by: George Hotz <[email protected]>
This commit is contained in:
Sieds Lykles
2025-02-13 12:54:15 +08:00
committed by GitHub
co-authored by George Hotz
parent 74742c018f
commit 095504d094
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/rewriter.py
+1 -1
View File
@@ -528,7 +528,7 @@ def full_graph_rewrite(sink:UOp, opts:Optional[Renderer]=None) -> UOp:
mulacc_unrolled)
else:
# new devectorize only for load/store
sink = graph_rewrite(sink, sym+devectorize_load_store)
sink = graph_rewrite(sink, sym+devectorize_load_store+mulacc_unrolled)
# final rules for the renderer (without sym)
sink = graph_rewrite(sink, symbolic_simple+get_late_rewrite_patterns(supported_ops, TRANSCENDENTAL>=2)+pm_render+extra_matcher)