An OpenCL 3D FFT for Molecular Dynamics Simulations on Multiple FPGAs
Silicon Therapeutics, 451 D St, Suite 205, Boston, MA, USA
arXiv:2009.12617 [cs.AR], (26 Sep 2020)
@misc{stewart2020opencl,
title={An OpenCL 3D FFT for Molecular Dynamics Simulations on Multiple FPGAs},
author={Lawrence C. Stewart and Carlo Pasoe and Brian W. Sherman and Martin Herbordt and Vipin Sachdeva},
year={2020},
eprint={2009.12617},
archivePrefix={arXiv},
primaryClass={cs.AR}
}
3D FFTs are used to accelerate MD electrostatic forces computations but are difficult to parallelize due to communications requirements. We present a distributed OpenCL 3D FFT implementation on Intel Stratix 10 FPGAs for grids up to 128^3. We use FPGA hardware features such as HBM2 memory and multiple 100 Gbps links to provide scalable memory accesses and communications. Our implementation outperforms GPUs for smaller FFTs, even without distribution. For 32^3 we achieve 4.4 microseconds on a single FPGA, similar to Anton 1 on 512 nodes. For 8 parallel pipelines (hardware limited), we reach the same performance both locally and distributed, showing that communications are not limiting the performance. Our FFT implementation is designed to be part of the electrostatic force pipeline of a scalable MD engine.
October 4, 2020 by hgpu
