DiffTune: Optimizing CPU Simulator Parameters with Learned Differentiable Surrogates

TL;DR

DiffTune introduces a method to learn accurate CPU simulator parameters by replacing the simulator with a differentiable surrogate, enabling gradient-based optimization to improve simulation accuracy using coarse measurements.

Contribution

It presents a novel approach to automatically tune CPU simulator parameters via differentiable surrogates, reducing reliance on expert knowledge and fine-grained measurements.

Findings

Learned parameters reduce simulation error.

Automatically tunes microarchitecture-specific parameters.

Improves accuracy over expert-provided values.

Abstract

CPU simulators are useful tools for modeling CPU execution behavior. However, they suffer from inaccuracies due to the cost and complexity of setting their fine-grained parameters, such as the latencies of individual instructions. This complexity arises from the expertise required to design benchmarks and measurement frameworks that can precisely measure the values of parameters at such fine granularity. In some cases, these parameters do not necessarily have a physical realization and are therefore fundamentally approximate, or even unmeasurable. In this paper we present DiffTune, a system for learning the parameters of x86 basic block CPU simulators from coarse-grained end-to-end measurements. Given a simulator, DiffTune learns its parameters by first replacing the original simulator with a differentiable surrogate, another function that approximates the original function; by making the surrogate differentiable, DiffTune is then able to apply gradient-based optimization techniques even when the original function is non-differentiable, such as is the case with CPU simulators. With this differentiable surrogate, DiffTune then applies gradient-based optimization to produce values of the simulator's parameters that minimize the simulator's error on a dataset of ground truth end-to-end performance measurements. Finally, the learned parameters are plugged back into the original simulator. DiffTune is able to automatically learn the entire set of microarchitecture-specific parameters within the Intel x86 simulation model of llvm-mca, a basic block CPU simulator based on LLVM's instruction scheduling model. DiffTune's learned parameters lead llvm-mca to an average error that not only matches but lowers that of its original, expert-provided parameter values.