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

add name to multi rewrite [pr] (#12346)

Browse Source
This commit is contained in:
qazal
2025-09-30 06:34:58 +03:00
committed by GitHub
parent 05275c9ec3
commit d95d018bb5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/schedule/multi.py
View File

@@ -241,4 +241,4 @@ multi_pm = PatternMatcher([
])+replace_allreduce
@track_rewrites()
def get_multi_map(big_sink:UOp) -> dict[UOp, UOp]: return graph_rewrite_map(big_sink, multi_pm)
def get_multi_map(big_sink:UOp) -> dict[UOp, UOp]: return graph_rewrite_map(big_sink, multi_pm, name="multi_pm")