SIMD vectorization for the Lennard-Jones potential with AVX2 and AVX-512 instructions

Hiroshi Watanabe, Koh M. Nakagawa

Research output: Contribution to journalArticlepeer-review

5 Citations (Scopus)

Abstract

This work describes the SIMD vectorization of the force calculation of the Lennard-Jones potential with Intel AVX2 and AVX-512 instruction sets. Since the force-calculation kernel of the molecular dynamics method involves indirect access to memory, the data layout is one of the most important factors in vectorization. We find that the Array of Structures (AoS) with padding exhibits better performance than Structure of Arrays (SoA) with appropriate vectorization and optimizations. In particular, AoS with 512-bit width exhibits the best performance among the architectures. While the difference in performance between AoS and SoA is significant for the vectorization with AVX2, that with AVX-512 is minor. The effect of other optimization techniques, such as software pipelining together with vectorization, is also discussed. We present results for benchmarks on three CPU architectures: Intel Haswell (HSW), Knights Landing (KNL), and Skylake (SKL). The performance gains by vectorization are about 42% on HSW compared with the code optimized without vectorization. On KNL, the hand-vectorized codes exhibit 34% better performance than the codes vectorized automatically by the Intel compiler. On SKL, the code vectorized with AVX2 exhibits slightly better performance than that with vectorized AVX-512.

Original languageEnglish
Pages (from-to)1-7
Number of pages7
JournalComputer Physics Communications
Volume237
DOIs
Publication statusPublished - 2019 Apr
Externally publishedYes

Keywords

  • AVX-512
  • AVX2
  • Molecular Dynamics Simulation
  • SIMD Vectorization
  • Xeon Phi

ASJC Scopus subject areas

  • Hardware and Architecture
  • Physics and Astronomy(all)

Fingerprint

Dive into the research topics of 'SIMD vectorization for the Lennard-Jones potential with AVX2 and AVX-512 instructions'. Together they form a unique fingerprint.

Cite this