Virtual Reservoir Acceleration for CPU and GPU: Case Study for Coupled Spin-Torque Oscillator Reservoir
Abstract
We provide high-speed implementations for simulating reservoirs described by $N$-coupled spin-torque oscillators. Here $N$ also corresponds to the number of reservoir nodes. We benchmark a variety of implementations based on CPU and GPU. Our new methods are at least 2.6 times quicker than the baseline for $N$ in range $1$ to $10^4$. More specifically, over all implementations the best factor is 78.9 for $N=1$ which decreases to 2.6 for $N=10^3$ and finally increases to 23.8 for $N=10^4$. GPU outperforms CPU significantly at $N=2500$. Our results show that GPU implementations should be tested for reservoir simulations. The implementations considered here can be used for any reservoir with evolution that can be approximated using an explicit method.
Cite
Text
De Jong et al. "Virtual Reservoir Acceleration for CPU and GPU: Case Study for Coupled Spin-Torque Oscillator Reservoir." NeurIPS 2023 Workshops: MLNCP, 2023.Markdown
[De Jong et al. "Virtual Reservoir Acceleration for CPU and GPU: Case Study for Coupled Spin-Torque Oscillator Reservoir." NeurIPS 2023 Workshops: MLNCP, 2023.](https://mlanthology.org/neuripsw/2023/jong2023neuripsw-virtual/)BibTeX
@inproceedings{jong2023neuripsw-virtual,
title = {{Virtual Reservoir Acceleration for CPU and GPU: Case Study for Coupled Spin-Torque Oscillator Reservoir}},
author = {De Jong, Thomas and Akashi, Nozomi and Taniguchi, Tomohiro and Notsu, Hirofumi and Nakajima, Kohei},
booktitle = {NeurIPS 2023 Workshops: MLNCP},
year = {2023},
url = {https://mlanthology.org/neuripsw/2023/jong2023neuripsw-virtual/}
}