浅水模式的显卡加速（40倍）

Lucian

原shallow water model的python版：https://github.com/maul1609/shallow-water-model-practical
将原模式翻译成c++，然后按c++代码写cuda代码给nvidia gpu运行，最后hipify成hip代码给amd gpu运行。https://github.com/YifeiLuDublin/shallow-water-model-practical

运行参数设置：
假设地球所有表平面高度为0，初始风场为西风，1440* 720 分辨率， 1分钟的积分间隔， 共积分4天。

运行时间：
Python: 18 min 23 s  @Ryzen 3600 单核
C++ 单核: 15 min 39 s @Ryzen 3600 单核
CUDA: 27 s @2080 super
HIP: 44 s @Vega 64


根据每小时输出的最大风速，判断出模式计算结果四版本相同

python版本的出图效果：https://imgur.com/a/8Ty6kpB

ljchen1989

火钳刘明，给楼主点赞~
请问楼主python和c++是用的单核嘛
nvidia gpu是啥型号？

Lucian

ljchen1989 发表于 2020-11-11 10:28
火钳刘明，给楼主点赞~
请问楼主python和c++是用的单核嘛
nvidia gpu是啥型号？

在帖子里更新了相关信息

ljchen1989

Lucian 发表于 2020-11-11 17:07
在帖子里更新了相关信息

嗯嗯，楼主辛苦，给楼主点赞
所以，好奇同等价位一颗金牌u和2080的效率对比

Lucian

ljchen1989 发表于 2020-11-11 19:45
嗯嗯，楼主辛苦，给楼主点赞
所以，好奇同等价位一颗金牌u和2080的效率对比

金牌u指xeon gold吗，如果单核的话估计跟我这个ryzen 3600单核差不多，顶多时间差距20%左右吧，如果要把核心全部用上的话得把python或者c++用openmp或者mpi重新改了

astiny

很强大！！