integrate CKKS with TPU kernels into HEIR

[JianmingTONG]

integrate CKKS into HEIR

[AlexanderViand-Intel]

Hi :) HEIR already has support for CKKS (there's a CKKS dialect, we have a --secret-to-ckks pass, and there's support for emitting OpenFHE code for CKKS. I think there's also an end-to-end example in the demo notebook). Feel free to edit & re-open this issue or create a new issue if there's a specific feature/capability that you're missing!

EDIT: re-opened as I just saw asraa update the title to be more specific about CKKS-on-TPU

[asraa]

yes! we're in a meeting right now but the intent for this was to integrate TPU kernels into HEIR which has a few routes:

• Lower through a shell API like jaxite (implement CKKS to jaxite and emit python code)
• Lower CKKS to polynomial, and then polynomial to Mosaic MLIR which can be compiled and scheduled for TPU (What is Mosaic jax-ml/jax#16855)

Will update more!