NVIDIA 推出了 cuDNN Graph API,这是其 cuDNN Frontend 中的一项新功能,允许开发者将计算定义为操作图。该 API 使 cuDNN 能够自动选择和优化这些图的执行引擎,并结合融合、自动调优和计划重用等技术。本教程通过构建和执行融合卷积、探索不同配置下的自动调优,以及集成 FP8 支持和 CUDA 图捕获等功能,演示了如何使用此 API,并与 PyTorch 的结果进行了验证。 AI
影响 通过允许开发者在 NVIDIA 的 cuDNN 库中定义和优化复杂的操作图,提高了 AI 计算效率。
排序理由 该条目描述了一个新的 API 及其在特定软件库 (cuDNN) 中的用法,属于工具范畴,而非核心前沿发布或重大行业事件。
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →