google / heir

A compiler for homomorphic encryption
https://heir.dev/
Apache License 2.0
289 stars 42 forks source link

Jupyter playground for HEIR #854

Open j2kun opened 1 month ago

j2kun commented 1 month ago

I created a proof of concept in https://github.com/j2kun/heir-play

It downloads the nightly build and then allows you to run %%heir-opt --flags in a cell and runs the passes and prints the output.

I think folks should try it out, and if it seems OK then I can publish it to PyPI and people can pip-install it and not need to clone the repo.

AlexanderViand-Intel commented 1 month ago

This is awesome! See https://github.com/j2kun/heir-play/pull/1 for some small fixes!

Can we integrate this into the main heir repo? It'd be nice to have this live in-tree + it'd look more "official" /trusted. That's assuming there isn't too much additional red tape for publishing a PyPI package from a google repo, though

j2kun commented 1 month ago

Yeah, I put it there before deciding where it should live. (Separate repo? In a "HEIR" GH org? In the main repo?)

AlexanderViand-Intel commented 1 month ago

It just occurred to me that we should add heir-translate, too, so that people can generate OpenFHE code and/or accelerator stuff to see what the outputs look like, or even to copy & paste them into something and actually run them.

j2kun commented 1 month ago

Added heir-translate. Last step I think is to migrate j2kun/heir-play to this repo and release it on PyPI