在 Tenstorrent Grayskull RISC-V 加速器上加速模板计算
分布式、并行与集群计算
2024-09-30 v1
摘要
RISC-V 指令集架构（ISA）近年来取得了惊人的增长，但在高性能计算（HPC）领域仍有待普及。尽管在 HPC 中采用 RISC-V CPU 解决方案可能尚需时日，但基于 RISC-V 的 PCIe 加速器提供了一个折中方案，使供应商既能受益于 RISC-V 的灵活性，又能融入现有系统。本文重点关注 Tenstorrent Grayskull PCIe RISC-V 加速器，该加速器构建于 Tensix 核之上，将数据移动与计算解耦。以 Jacobi 迭代法为载体，我们探讨了模板计算在 Grayskull e150 上的适用性。我们探讨了在加速器上组织这些代码的最佳实践，并证明 e150 提供了与 Xeon Platinum CPU 相当的性能（尽管是 BF16 与 FP32 相比），但 e150 的能耗约为后者的五分之一。在四个 e150 上，我们获得了约四倍的 CPU 性能，同样能耗约为五分之一。
引用
@article{arxiv.2409.18835,
title = {Accelerating stencils on the Tenstorrent Grayskull RISC-V accelerator},
author = {Nick Brown and Ryan Barton},
journal= {arXiv preprint arXiv:2409.18835},
year = {2024}
}
备注
Author accepted version, to appear in proceedings of the international workshop on RISC-V for HPC