适用于Python的HPC基准
这是一组基准测试,用于测试使用Python前端的各种计算后端的顺序CPU和GPU性能。
具体来说,我们想测试哪种高性能后端最适合地球物理(基于有限差分)的模拟。
内容
常问问题
为什么?
科学的Python生态系统正在蓬勃发展,但是Python中的高性能计算还不是真正的事情。 我们尝试来更改此,但是我们应该使用哪个后端进行计算?
Python前端到高性能后端的开发需要大量的时间和资源,但是这些通常是为深度学习量身定制的。 我们想了解一下,通过(滥用)这些库进行地球物理建模,我们是否可以从这些进展中获利。
为什么基准看起来如此怪异?
这些或多或少是逐字记录副本(即物理模型的实际部分)。 大多数地球系统和气候模型组件都基于有限差分方案来计算导数。 这可以通过数组的索引移位(例如0.5 * (arr[1:] + arr[:-1]) , arr在每个点的一
1