跳到正文
arXiv 物理· Seunghyun Joo·· 13 小时前AI 评分40

CuAD:面向计算流体力学离散伴随的 CUDA 内核源到源自动微分工具

CuAD: Source-to-Source Automatic Differentiation of CUDA Kernels for Discrete Adjoints in Computational Fluid Dynamics

AI 导读

CuAD 是一款源到源自动微分工具,可读取 CUDA 内核并输出其切线与伴随版本的普通 CUDA 源码,反向模式以 GPU 线程为组织单位。

来源:arXiv 物理 · arxiv.org