An MLIR-based toolchain for homomorphic encryption compilers. Read the docs at the HEIR website.
For more information on MLIR, see the MLIR homepage.
Pip install the heir_py
package
pip install heir_py
Then run an example:
from heir import compile
from heir.mlir import F32, I16, I64, Secret
@compile() # defaults to scheme="bgv", OpenFHE backend, and debug=False
def func(x: Secret[I16], y: Secret[I16]):
sum = x + y
diff = x - y
mul = x * y
expression = sum * diff + mul
deadcode = expression * mul
return expression
foo.setup()
enc_a = foo.encrypt_a(7)
enc_b = foo.encrypt_b(8)
result_enc = foo.eval(enc_a, enc_b)
result = foo.decrypt_result(result_enc)
print(
f"Expected result for `func`: {func.original(7,8)}, FHE result:"
f" {result}"
)
This will compile the function above using the BGV scheme to machine code via
the OpenFHE backend.
Then calling the function will encrypt the inputs, run the function, and return
the decrypted result. The function call foo(7, 8)
runs the entire
encrypt-run-decrypt flow for ease of testing.
The python package heir_py
ships with the heir-opt
and heir-translate
. If
you install via virtualenv
, the binaries will be in your venv/bin
.
venv/bin/heir-opt --help
Backend Library | BGV | BFV | CKKS | CGGI |
---|---|---|---|---|
OpenFHE | ✅ | ✅ | ✅ | ❌ |
Lattigo | ✅ | ✅ | ✅ | ❌ |
tfhe-rs | ❌ | ❌ | ❌ | ✅ |
Jaxite | ❌ | ❌ | ❌ | ✅ |
Note some backends do not support all schemes.
There are many ways to contribute to HEIR:
- Come to our monthly meetings to discuss active work on HEIR and future project directions. The meetings are recorded and posted to our blog and YouTube channel.
- Come to our weekly office hours for informal discussions and debugging help.
- Ask questions or discuss feature ideas in the
#heir
channel on the FHE.org discord. - Work on an issue marked "good first issue" or browse issues labeled by topic.
- Help us understand new FHE research: either
- Read a paper tagged under research synthesis and summarize the novel techniques that could be ported to HEIR.
- File new issues under research synthesis to alert us of papers that should be investigated and incorporated into HEIR.
The HEIR project can be cited in in academic work through following entry:
@Misc{HEIR,
title={{HEIR: Homomorphic Encryption Intermediate Representation}},
author={HEIR Contributors},
year={2023},
note={\url{https://github.com/google/heir}},
}
This is not an officially supported Google product.