@gpuwaster: GPU Grind 第61/100篇:正在学习这场GTC 2020讲座:开发CUDA内核将Tensor Core推向绝对极限…

X AI KOLs Timeline 事件

摘要

作者在其共100部分的GPU学习系列中,回顾了Andrew Kerr在GTC 2020上的一场讲座,内容是关于开发高性能CUDA内核以将NVIDIA A100的Tensor Core性能推向极致,探讨了相关技术以及原生CUDA与CUTLASS之间的取舍。

GPU Grind 第61/100篇 正在学习这场GTC 2020讲座:开发CUDA内核,将NVIDIA A100上的Tensor Core推向绝对极限 这个标题确实很有趣,因为它正是我目前正在尝试做的事情。这是CUTLASS团队的Andrew Kerr所举办的讲座,回顾了在Ampere架构上优化GEMM的策略,解释了mma指令、ldmatrix指令和cp.async的工作原理,以及创建一个良好的内存管线需要什么。 我开始看视频时想寻找新的技巧,但发现这些全是我已经在做的事情(不过他们可能做得更好,特别是swizzling部分)。 我从中得出的主要结论是,使用CUTLASS编写这样的内核会容易得多……这差不多就是讲座的重点,毕竟它是CUTLASS团队讲的。但我觉得如果不用原生CUDA来写,我就学不到那么多东西了。我是不是在自讨苦吃? 链接:https://developer.nvidia.com/gtc/2020/video/s21745-vid…
查看原文
查看缓存全文

缓存时间: 2026/07/12 10:53

61/100 的 GPU 磨炼

正在观看 GTC 2020 的讲座:开发 CUDA 内核,将 NVIDIA A100 上的 Tensor Core 推向极限

这个标题确实很有意思,因为正好是我正在做的事。这是 CUTLASS 团队 Andrew Kerr 的演讲,回顾了在 Ampere 上优化 GEMM 的策略,解释了 mma 指令、ldmatrix 指令和 cp.async 的工作原理,以及构建良好内存流水线所需的要素。

开始看视频时,我本在寻找新技巧,但发现都是我已经在做的事情(不过他们可能做得更好,尤其是在交错排列部分)。

我从中学到的主要结论是:用 cutlass 编写这样的内核会轻松得多……这大概也是讲座的目的吧——毕竟是 cutlass 团队的演讲,但我感觉如果不用原生 CUDA 来做,我学到的东西就没那么多。我这是在自讨苦吃吗?

链接:https://developer.nvidia.com/gtc/2020/video/s21745-vid…


GTC 2020:开发 CUDA 内核,将 NVIDIA A100 上的 Tensor Core 推向极限

来源:https://developer.nvidia.com/gtc/2020/video/s21745-vid

点击“立即观看”后,系统会提示您登录或加入。

立即观看 (https://developer.nvidia.com/gtc/2020/video/s21745)


GTC 2020:开发 CUDA 内核,将 NVIDIA A100 上的 Tensor Core 推向极限 (https://developer.nvidia.com/gtc/2020/video/s21745) 点击“立即观看”以登录或加入 NVIDIA 开发者计划。

立即观看 (https://developer.nvidia.com/gtc/2020/video/s21745)

开发 CUDA 内核,将 NVIDIA A100 上的 Tensor Core 推向极限

Andrew Kerr,NVIDIA

GTC 2020

NVIDIA Ampere GPU 架构通过将 Tensor Core 的混合精度数学吞吐量翻倍来提升性能,同时还增加了对双精度、Tensor Float 32 和 bfloat16 数据类型的支持。我们将介绍如何使用 A100 上的 Tensor Core 实现高性能 CUDA 内核,应用寄存器分块、软件流水线和精心构建的内存布局以避免存储体冲突等技术。然后,我们将介绍 CUTLASS 中用于编程 Tensor Core 的抽象以及其他新特性。本讲座面向高级 CUDA C++ 程序员,他们渴望编写将 Tensor Core 推向峰值性能的内核。我们建议您先回顾以往关于该主题的演讲,例如 CUTLASS 简介(GTC 2018)和 CUTLASS 中的 Volta Tensor Core 编程(GTC 2019)。


查看更多 GTC 2020 内容 (https://developer.nvidia.com/events/recordings/gtc-d-2020)

thom✨ (@gpuwaster): 60/100 的 GPU 磨炼

正在学习斯坦福 cs149 课程的第 4 讲,内容涵盖并行程序创建的基础知识,从阿姆达尔定律、问题分解为任务(由程序员完成)、任务分配(通常由编译器完成),以及 ISPC 中如何实现这些。

相似文章