基于 TPU 的高分辨率成像
计算工程、金融与科学
2019-12-18 v1 分布式、并行与集群计算
计算物理
地球物理
摘要
人工智能（AI）的快速发展正催生出针对深度学习优化的新一代硬件加速器。其中部分设计具有足够的通用性，可使其用于 AI 之外其他计算密集型任务。云端张量处理单元（TPU）即为一例。在此，我们展示一种在 Cloud TPU 上使用 TensorFlow 实现称为全波形反演的高分辨率成像技术的新方法。高阶数值模板利用了 Cloud TPU 提供的高效矩阵乘法，而 halo 交换得益于专用的高速片间互连。与 Tesla V100 图形处理单元相比性能具有竞争力，并展现出对未来计算与内存密集型成像应用的前景。
引用
@article{arxiv.1912.08063,
title = {High-resolution imaging on TPUs},
author = {Fantine Huot and Yi-Fan Chen and Robert Clapp and Carlos Boneti and John Anderson},
journal= {arXiv preprint arXiv:1912.08063},
year = {2019}
}
备注
12 pages, 6 figures, submitted to ISC High Performance 2020