From d95d018bb5ab2c36bd05d7ef2d25e096db03c1a9 Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Tue, 30 Sep 2025 06:34:58 +0300 Subject: [PATCH] add name to multi rewrite [pr] (#12346) --- tinygrad/schedule/multi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/schedule/multi.py b/tinygrad/schedule/multi.py index a1cfe18c1a..b6623e0f14 100644 --- a/tinygrad/schedule/multi.py +++ b/tinygrad/schedule/multi.py @@ -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")