Possibility to align a split. - #9371

Draft

mcourteaux wants to merge 38 commits into

mainfrom

mcourteaux/aligned-split

Commits on Aug 25, 2026

WIP: shared BoundsTracker for BoundConstantExtentLoops/BoundSmallAllocations/AllocationBoundsInference

mcourteaux
and
claude
committed
5f684b3
Fix use-after-free in BoundConstantExtentLoops's aggressive extent lookup

mcourteaux
and
claude
committed
29f1275
Migrate LowerWarpShuffles onto BoundsTracker

mcourteaux
and
claude
committed
681d1f1
Gather Let/Assert context in SimplifyCorrelatedDifferences via BoundsTracker

mcourteaux
and
claude
committed
9b95241

Commits on Aug 26, 2026

Bound a min-clamped ceil-div minus its unclamped multiple
mcourteaux
committed
4d7cd44
Use loop monotonicity to bound an unrollable tile exactly

mcourteaux
and
claude
committed
2137977
Move one simplification call in lowering.

mcourteaux
and
abadams
committed
a5d88de
Move printing of the final simplification and call it 'after reaching conceptual stmt'
mcourteaux
committed
cf3e022
PoC: Aligned split directive.
mcourteaux
committed
d978880
Add enable backtraces to async_copy_chain as I have found it to be stalled/deadlocked several times but couldn't debug it.
mcourteaux
committed
891e8e4
Restore likely_if_innermost for ShiftInwards.
mcourteaux
committed
cc30125
Add Python binding

mcourteaux
and
claude
committed
c903c8c
Add serialization.

mcourteaux
and
claude
committed
fad568a
Fix the incorrectly assumed fast-path for this aligned splits.

mcourteaux
and
claude
committed
973eecb
Add three rfactor + aligned split tests.

committed
4c3dd97
Add test for nested aligned splits.

mcourteaux
and
claude
committed
6b38cff
Test varying tail strategies for nested aligned splits.

mcourteaux
and
claude
committed
41d3858
Documentation for the aligned split.

mcourteaux
and
claude
committed
7cece55
Fix ShiftInwardsAndBlend, RoundUpAndBlend. Claude rederived the masks those blend operations in case of aligned splits.

mcourteaux
and
claude
committed
c929d18
Add simplifier rules for broadcast() <= ramp() && ramp() <= broadcast().

mcourteaux
and
claude
committed
7bfc6f7
Test simple aligned split in an RVar.

mcourteaux
and
claude
committed
7d1867c
WIP checkpoint: aligned-split debug tracing, simplifier rules, 6x6 compute_at test
mcourteaux
committed
bf05e6c
Reduce aligned-split compute_at mux test to 3x3 and fix its checks
mcourteaux
committed
3823f18
Give producers a tile-aligned region inside an aligned split
mcourteaux
committed
9d5a33f
Starting point.
mcourteaux
committed
f0c047b
Fix bad merge.
mcourteaux
committed
9ee4682
Add a test for an aligned split feeding loop partitioning
mcourteaux
committed
514936e
Let can_prove predicates use the simplifier's known facts

mcourteaux
and
claude
committed
990ef0b
Make fact lookup aware of comparison direction and strictness

mcourteaux
and
claude
committed
482af79
Don't re-enter fact-driven rewrite rules from inside a can_prove

mcourteaux
and
claude
committed
caba268
Express the can_prove re-entry guard as a depth limit

mcourteaux
and
claude
committed
9ba560f

Commits on Aug 27, 2026

Add a non-recursive known_true predicate for rewrite rules

mcourteaux
and
claude
committed
55827b5
Merge branch 'mcourteaux/can-prove-facts' into mcourteaux/aligned-split
mcourteaux
committed
7215b29
Fix parenthesis of Simplify_Div.
mcourteaux
committed
a983f5a
Guard against can_prove recursion at its source

mcourteaux
and
claude
committed
fa4a807
Fall back to fact lookup at the can_prove depth cap

mcourteaux
and
claude
committed
52a5e87
Use a direct fact lookup at the can_prove depth cap, not a tree walk

mcourteaux
and
claude
committed
032f724
Merge branch 'mcourteaux/can-prove-facts' into mcourteaux/aligned-split
mcourteaux
committed
1df274a