Automatic Generation of Distributed-Memory Mappings for Tensor Computations
Martin Kong, Raneem Abu-Yosef, Atanas Rountev, Ponnuswamy Sadayappan · 2023
While considerable research has been directed at automatic parallelization for shared-memory platforms, little progress has been made in automatic parallelization schemes for distributed-memory systems. We introduce an innovative approach to automatically produce distributed-memory parallel code for an important subclass of affine tensor computations common to Coupled Cluster (CC) electronic structure methods, neuro-imaging applications, and deep learning models.