有限元求解器搬进可微图程序:同一离散下,单块H200每步比32核CPU快181倍
共轭传热与浮力流的有限元求解器被改写为自动微分框架中的固定权重图程序,装配算子与代数多重格点都按稀疏图执行,不做任何训练。在离散、网格、时间步和容差完全相同的条件下,同一问题在单块H200上每步最多比32核CPU快181倍,压力求解单独快18.7倍。反向模式梯度与中心差分一致,隐式微分让一次梯度的代价稳定在约4.9次前向求解。
共轭传热与浮力流的有限元求解器被改写为自动微分框架中的固定权重图程序,装配算子与代数多重格点都按稀疏图执行,不做任何训练。在离散、网格、时间步和容差完全相同的条件下,同一问题在单块H200上每步最多比32核CPU快181倍,压力求解单独快18.7倍。反向模式梯度与中心差分一致,隐式微分让一次梯度的代价稳定在约4.9次前向求解。
已读完全部报道