668: Push-down of arithmetics before source operations - #967

Open

b-m-f wants to merge 44 commits into

daphne-project:mainfrom

b-m-f:668-pushdown-arithmetics

Commits on Aug 18, 2025

[668]: Push down arithmetics into fill operation for EWAdd

bob@sodawa.com
authored and
Maximilian Ehlers
committed
5602b91
[668]: Only store 1 variable of the FillOp. Add TODO for updating docs.

bob@sodawa.com
authored and
Maximilian Ehlers
committed
5526966
[668]: Marks FillOp as pure operation. Removes manual operation removal in FillOp arithmetic push down canonicalization

bob@sodawa.com
authored and
Maximilian Ehlers
committed
94326b2
[668]: adds lit based mlir test case for pushdown arithmetics

bob@sodawa.com
authored and
Maximilian Ehlers
committed
2481db5
[668]: use argument names to access fill operation arguments

bob@sodawa.com
authored and
Maximilian Ehlers
committed
e8c0d07
[668]: adds fill rewrite for EwSubOp,EwMulOp,EwDivOp based on EwAddOp implementation
Maximilian Ehlers
committed
977a8fc
[668]: adds PushDown for RandomMatrixOp+EwAddOp. Marks RandomMatrix as pure to allow dead code elimination
Maximilian Ehlers
committed
154b829
[668]: extract pushdown logic into seperate functions
Maximilian Ehlers
committed
00e8a49
[668]: enables canonicalization for EwMod, EwLog, EwPow and implements Pushdown arithmetics for EwMod and EwPow
Maximilian Ehlers
committed
f48c5e4
[668]: renames tests to include both operations in filename
Maximilian Ehlers
committed
ea21abe
[668]: remove broken test for FillEwLog and adds tests for RandEwAdd
Maximilian Ehlers
committed
8f7ef5d
[668]: Adds pushdown arithmetics for Rand + EwMul,EwDiv,EwSub,EwPow
Maximilian Ehlers
committed
ce39cd8
[668]: Enables Pushdown for Rand and Fill with EwLog
Maximilian Ehlers
committed
ca81ae2
[668]: fix swapped height and with on fill pushdown
Maximilian Ehlers
committed
1d0f3cb
[668]: add scripts test cases for fill EwAdd, EwSub
Maximilian Ehlers
committed
b763e81
[668]: adds more tests for binaryOp Fill pushdown
Maximilian Ehlers
committed
d12cb46
[668]: adds tests for pushdown in rand function
Maximilian Ehlers
committed
86b6b6d
[668]: removes cases for function (matrix) on rhs of binary op
Maximilian Ehlers
committed
e97a919
[668]: removes comments now reflected in tests
Maximilian Ehlers
committed
d0767bb
[668]: updates distributed_4 test to have two different istantiations of the randomly generated matrix
Maximilian Ehlers
committed
6362f4e
[668]: removes unused variable. Sets correct type for new EwLog Operation on pushdown
Maximilian Ehlers
committed
b058378
[668]: updates rand pushdown mlir tests
Maximilian Ehlers
committed
8cbc2cb
[668]: adds pushdown for fill operation with unary arithmetics
Maximilian Ehlers
committed
b621259
[668]: adds pushdown for randMatrix operation and unary arithmetics
Maximilian Ehlers
committed
f6e6dec
[668]: disable incorrect random ewAbs pushdown
Maximilian Ehlers
committed
c3abdf4
[668]: adds test cases for seq operation pushdown. seq + div seems to be buggy - does not output doubles
Maximilian Ehlers
committed
39981c3
[668]: fix Fill + Sub pushdown test results
Maximilian Ehlers
committed
976a8bc
[668]: enables pushdown for seq operation and add,sub,mul,div
Maximilian Ehlers
committed
7065f40
[668]: refactor Canonicalizations into template methods
Maximilian Ehlers
committed
72e3213
[668]: renames script test files to not reference internal details
Maximilian Ehlers
committed
ada17d6
[668]: adds Traits to handle the different combinations of Operations for PushDown
Maximilian Ehlers
committed
d188bc2
[668]: removes nonlinear rand pushdowns and updates the special randAbs case.
Maximilian Ehlers
committed
4805bb9
[668]: removes unused test cases as rand does not support non linear pushdown
Maximilian Ehlers
committed
23787cc
[668]: fixes tests, removes seqEwDiv which is not suitable for pushDown in integer space, adds docs to tryPushDown
Maximilian Ehlers
committed
a56073a
[668: fixes width and height confusion on rand + unary/binary pushdown
Maximilian Ehlers
committed
43cbc81
[668]: makes sure rand and ewAbs work when mix = max, makes sure the value checks are performed safely
Maximilian Ehlers
committed
8e39be1
[668]: remove pushdown for rand + EwMul/EwDiv if the range delimiters are integers
Maximilian Ehlers
committed
b6f1d19
[668]: fixes seq with EwDiv pushdown and adds more tests for it
Maximilian Ehlers
committed
87c43f3
[668]: removes superfluous PushDown trait
Maximilian Ehlers
committed
573ef68
[668]: removes intermediate tryPushDown method to reduce code complexity
Maximilian Ehlers
committed
c14f8a9
[668]: activates pushdown for EwMinus
Maximilian Ehlers
committed
a92a27f
[668]: renames trait PushDownWithIntervalUpdate to PushDownIncrementUpdate
Maximilian Ehlers
committed
8dfca70
[668]: replace create and replaceOp with replaceOpWithNewOp()
Maximilian Ehlers
committed
58f19f9

Commits on Aug 19, 2025

[668]: update MR to work with changes from b55d1a88358196f8070622cf6a59a26e515c0295
Maximilian Ehlers
committed
31b30c7