[Mlir-commits] [mlir] [mlir][Conversion] Raise SCF to Affine. (Ported from Polygeist) (PR #158267)

[William Moses]

wsmoses wrote:

First of all, thank you so much for helping with this -- it would be awesome to see this upstream (as many many projects would benefit from the polygeist raising and other transformations).

The most up to date version of the code currently lives here (and contains raising of not just loops, but memory, etc): https://github.com/EnzymeAD/Enzyme-JAX/blob/main/src/enzyme_ad/jax/Passes/AffineCFG.cpp

At this point might I recommend perhaps we just add a separate polygeist dialect upstream and make the requisite pass under that?

cc @pangoraw @ftynse @ivanradanov 

https://github.com/llvm/llvm-project/pull/158267