A Method for Efficient Heterogeneous Parallel Compilation: A Cryptography Case Study

TL;DR

This paper presents a new MLIR-based dialect called hyper that optimizes data management and parallel computation in heterogeneous systems, demonstrated through a cryptography compiler prototype achieving significant speedups.

Contribution

Introduction of the hyper dialect for efficient heterogeneous compilation and a cryptography compiler prototype showcasing improved performance over existing models.

Findings

Achieved average speedups of 1.93x for SHA-1, 1.18x for MD5, and 1.12x for SM3.

Demonstrated the hyper dialect's ability to harness heterogeneous hardware effectively.

Provided a unified compilation framework for diverse hardware architectures.

Abstract

In the era of diminishing returns from Moores Law, heterogeneous computing systems have emerged as a vital approach to enhance computational efficiency. This paper introduces a novel MLIR-based dialect, named hyper, designed to optimize data management and parallel computation across diverse hardware architectures. The hyper dialect abstracts the complexities of heterogeneous computing by providing a unified compilation framework that efficiently schedules tasks and manages data communication. To demonstrate its capabilities, we present HETOCompiler, a cryptography-focused compiler prototype that implements multiple hash algorithms and enables their execution on heterogeneous systems. The proposed approach achieves performance improvements over existing programming models for heterogeneous computing (OpenCL), offering an average speedup of 1.93x, 1.18x, and 1.12x for SHA-1, MD5, and SM3 algorithms, respectively. Our findings highlight the potential of the hyper dialect in harnessing the full computational power of heterogeneous devices, advancing the field of compiler design for heterogeneous systems.