https://github.com/joker-eph commented: That looks good from an MLIR point of view, I don't understand NTT enough to know if the patterns are correct though. So LG if you're owning the maths correctness :) https://github.com/llvm/llvm-project/pull/93132