Merge branch 'master' into late_decomp_is_late

move late decomp after new style
2026-08-15 22:58:28 +00:00 · 2026-06-24 21:57:10 -07:00 · 2026-06-24 19:30:32 -07:00
1 changed files with 5 additions and 7 deletions
@@ -127,18 +127,16 @@ def full_rewrite_to_sink(ast:UOp, ren:Renderer, optimize:bool=True) -> UOp:
  # do memory coalesing (late)
  sink = memory_coalesing(sink, ren)

-  # instruction selection decompositions
-  pm_decomp = pm_decomp+\
-    get_late_rewrite_patterns(supported_ops, bool(DISABLE_FAST_IDIV))+\
-    get_transcendental_patterns(supported_ops, TRANSCENDENTAL>=2)
-  sink = graph_rewrite(sink, pm_decomp, ctx=ren, name="late decompositions")
-
  # this is new style (TODO: this should all be removed)
  sink = graph_rewrite(sink, pm_render, name="pm_render gep/stack")
  sink = graph_rewrite(sink, pm_index_is_shrink, name="index is shrink")
  sink = graph_rewrite(sink, pm_remove_vec_dtypes, name="transform to new style")

-  # move gates from unrenderable INVALID where
+  # late decomps + move gates from unrenderable INVALID where
+  pm_decomp = pm_decomp+\
+    get_late_rewrite_patterns(supported_ops, bool(DISABLE_FAST_IDIV))+\
+    get_transcendental_patterns(supported_ops, TRANSCENDENTAL>=2)
+  sink = graph_rewrite(sink, pm_decomp, ctx=ren, name="late decompositions")
  sink = graph_rewrite(sink, pm_move_gates_from_index, name="move gates from index")

  # final rules for the renderer (without sym)
Author	SHA1	Message	Date
George HotzandGitHub	00b9be45d4	Merge branch 'master' into late_decomp_is_late	2026-06-24 21:57:10 -07:00
geohot	1b994094c2	move late decomp after new style	2026-06-24 19:30:32 -07:00