HEIR: A Universal Compiler for Homomorphic Encryption

TL;DR

HEIR is a versatile compiler framework built on MLIR that unifies homomorphic encryption techniques, enabling easier optimization, benchmarking, and integration across software and hardware platforms.

Contribution

This paper introduces HEIR, a comprehensive and adaptable compiler platform for homomorphic encryption that supports diverse techniques and simplifies research and development.

Findings

HEIR successfully ported a large portion of HE literature.

HEIR can handle complex and diverse programs.

HEIR is becoming a standard tool in HE research and industry.

Abstract

This work presents Homomorphic Encryption Intermediate Representation (HEIR), a unified approach to building homomorphic encryption (HE) compilers. HEIR aims to support all mainstream techniques in homomorphic encryption, integrate with all major software libraries and hardware accelerators, and advance the field by providing a platform for research and benchmarking. Built on the MLIR compiler framework, HEIR introduces HE-specific abstraction layers at which existing optimizations and new research ideas may be easily implemented. Although many HE optimization techniques have been proposed, it remains difficult to combine or compare them effectively. HEIR provides a means to effectively explore the space of HE optimizations. HEIR addresses the entire HE stack and includes support for various frontends, including Python. The contribution of this work includes: (1) We introduce HEIR as a framework for building HE compilers. (2) We validate HEIR's design by porting a large fraction of the HE literature to HEIR, and we argue that HEIR can tackle more complicated and diverse programs than prior literature. (3) We provide evidence that HEIR is emerging as the de facto HE compiler for academic research and industry development.