老Fortran大气物理代码搬上GPU:四个大模型各写一版JAX,最快一版比串行Fortran快约357倍
四个大模型把Kessler暖雨微物理的Fortran代码改写成GPU上的JAX,输出与Fortran参考几乎一致,但GPU性能相差很大,最快一版在缓存GPU执行下比串行Fortran快约357倍。
四个大模型把Kessler暖雨微物理的Fortran代码改写成GPU上的JAX,输出与Fortran参考几乎一致,但GPU性能相差很大,最快一版在缓存GPU执行下比串行Fortran快约357倍。
已读完全部报道