CoMoNM: A Cost Modeling Framework for Compute-Near-Memory Systems
Preprint
Abstract
Compute-Near-Memory (CNM) systems offer a promising approach to mitigate the von Neumann bottleneck by bringing computational units closer to data. However, optimizing for these architectures remains challenging due to their unique hardware and programming models. Existing CNM compilers often rely on manual programmer annotations for offloading and optimizations. Automating these decisions by exploring the optimization space, common in CPU/GPU systems, is difficult for CNMs as constructing and navigating the transformation space is tedious and time consuming. This is particularly the case during system-level design, where evaluation requires time-consuming simulations. To address this, we present CoMoNM, a generic cost modeling framework for CNM systems for execution time estimation in milliseconds. It takes a high-level, hardware-agnostic application representation, target system specifications, and a mapping specification as input and estimates the execution time for the given application on the target CNM system. We show how CoMoNM can be seamlessly integrated into state-of-the-art CNM compilers, providing improved offloading decisions. Evaluation on established benchmarks for CNM shows estimation errors within 7.80% and 2.99%, when compared to the real UPMEM CNM system and Samsung's HBM-PIM simulator. Notably, CoMoNM delivers estimates seven orders of magnitude faster compared to the UPMEM and HBM-PIM simulators.Keywords
All Related Versions
This publication has 20 references indexed in Scilit:
- CIMFlow: An Integrated Framework for Systematic Design and Evaluation of Digital CIM Architectures2025
- SADIMM: Accelerating Sparse Attention Using DIMM-Based Near-Memory ProcessingIEEE Transactions on Computers, 2024
- SimplePIM: A Software Framework for Productive and Efficient Processing-in-Memory2023
- SongC: A Compiler for Hybrid Near-Memory and In-Memory Many-Core ArchitectureIEEE Transactions on Computers, 2023
- XLA-NDP: Efficient Scheduling and Code Generation for Deep Learning Model Training on Near-Data Processing MemoryIEEE Computer Architecture Letters, 2023
- OCC: An Automated End-to-End Machine Learning Optimizing Compiler for Computing-In-MemoryIEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems, 2021
- ComPRIMe: A Compiler for Parallel and Scalable ReRAM-based In-Memory Computing2019
- Streaming End-to-end Speech Recognition for Mobile Devices2019
- Transparent Offloading and Mapping (TOM): Enabling Programmer-Transparent Near-Data Processing in GPU Systems2016
- The missing memristor foundNature, 2008