Investigate "Bridging Usability and Performance: A Tensor Compiler for Autovectorizing Homomorphic Encryption"
- Dominant language
- MLIR
- Stars
- 906
- Forks
- 171
- Avg merge
- 4d 12h
- Merged PRs (30d)
- 32
Description
This is Edward Chen's Rotom paper http://eprint.iacr.org/2025/1319
We had talked about adding HEIR as a backend to this tool. I think we should also determine what work would be required to port it as well.
> Edward Chen, Carnegie Mellon University
> Fraser Brown, Carnegie Mellon University
> Wenting Zheng, Carnegie Mellon University
>
> Abstract
> Homomorphic encryption (HE) offers strong privacy guarantees by enabling computation over encrypted data. However, the performance of tensor operations in HE is highly sensitive to how the plaintext data is packed into ciphertexts. Large tensor programs introduce numerous possible layout assignments, making it both challenging and tedious for users to manually write efficient HE programs.
>
> In this paper, we present Rotom, a compilation framework that autovectorizes tensor programs into optimized HE programs. Rotom systematically explores a wide range of layout assignments, applies state-of-the-art optimizations, and automatically finds an equivalent, efficient HE program. At its core, Rotom utilizes a novel, lightweight ApplyRoll layout conversion operator to easily modify the underlying data layouts and unlock new avenues for performance gains. Our evaluation demonstrates that Rotom scalably compiles all benchmarks in under 5 minutes, reduces rotations in manually optimized protocols by up to 4×, and achieves up to 80× performance improvement over prior systems.
Contributor guide
Assessment
This issue has not been assessed yet.