第1节:现代GPU硬件架构精讲
文章目录前言一、GPU vs CPU为什么GPU适合并行计算二、GPU的整体架构从芯片到核心2.1 GPU的层级结构2.2 A100/H100芯片架构图三、SM内部架构详解3.1 SM流式多处理器内部结构3.2 一个SM的详细数据以A100为例四、Warp最小的执行单元4.1 Warp的内部结构4.2 Warp的执行过程五、计算核心的进化CUDA Core → Tensor Core5.1 CUDA Core vs Tensor Core 对比图5.2 各代Tensor Core演进六、内存层次结构6.1 GPU内存层次图七、从硬件到软件的映射7.1 CUDA线程层次与硬件映射7.2 硬件资源限制示意图八、NVIDIA核心性能参数全景图 编程调优的关键洞察九、动手实践9.1 查询你的GPU信息9.2 编译运行十、本节总结10.1 核心要点回顾10.2 下节预告十一、参考资料十二、面试真题2024-2026Q1解释GPU的SIMT模型和CPU的SIMD有什么区别Q2什么是WarpWarp的大小为什么是32Q3什么是Occupancy如何计算和优化Q4Tensor Core和CUDA Core有什么区别分别在什么场景使用Q5SM上同时能跑多少个线程受什么限制Q5SM上同时能跑多少个线程受什么限制前言不懂硬件写不出高性能CUDA代码很多初学者学CUDA上来就写核函数结果发现同样的算法别人比我快10倍还不知道为什么。原因很简单你不了解GPU这个“合作伙伴”的脾气秉性。就像你要用好一个人得知道他是急性子还是慢性子、擅长数学还是语文。GPU也一样——它的架构设计直接决定了什么样的代码跑得快什么样的代码跑得慢。今天我们就从硬件层面彻底搞懂现代GPU的工作原理。一、GPU vs CPU为什么GPU适合并行计算先看一张经典的对比图CPUGPU少数强大核心数千个简单核心复杂控制单元简单控制单元大容量缓存小容量缓存适合串行任务适合并行任务一句话总结CPU是博士一个顶十个擅长处理复杂问题GPU是小学生单个能力弱但一万人一起上干活比博士快这就是所谓的延迟优化 vs 吞吐优化。二、GPU的整体架构从芯片到核心2.1 GPU的层级结构现代GPU是一个复杂的多层结构层级说明GPC图形处理簇GPU的最高层级划分每个GPC包含多个TPCTPC纹理处理簇每个TPC包含1-2个SMSM流式多处理器最核心的计算单元我们编程直接打交道的对象2.2 A100/H100芯片架构图以NVIDIA A100Ampere架构为例A100关键数据108个SM40MB L2缓存40/80GB HBM2显存1.6TB/s显存带宽三、SM内部架构详解3.1 SM流式多处理器内部结构SM是GPU最核心的计算单元我们直接看它的内部构造3.2 一个SM的详细数据以A100为例组件数量说明Warp调度器4个每个时钟周期可调度2条指令CUDA Core (FP32)64个单精度浮点运算CUDA Core (INT32)64个整数运算FP64 Core32个双精度浮点运算部分型号减半Tensor Core4个第四代Tensor Core寄存器文件256KB65536个32位寄存器共享内存/L1164KB可配置为共享内存或L1缓存最大线程数2048个同时驻留的线程数最大Warp数64个同时驻留的warp数最大线程块数32个同时驻留的block数四、Warp最小的执行单元4.1 Warp的内部结构┌──────────────────────────────────────────────────────────────────┐ │ 一个Warp32个线程 │ │ │ │ 线程ID: 0 1 2 3 4 5 6 7 8 9 10 11 12 13 │ │ ┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐ │ │T0 ││T1 ││T2 ││T3 ││T4 ││T5 ││T6 ││T7 ││T8 ││T9 ││T10││T11││T12││T13│ │ └───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘ │ │ │ 线程ID:14 15 16 17 18 19 20 21 22 23 24 25 26 27 │ │ ┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐┌───┐ │ │T14││T15││T16││T17││T18││T19││T20││T21││T22││T23││T24││T25││T26││T27│ │ └───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘└───┘ │ │ │ 线程ID:28 29 30 31 │ │ ┌───┐┌───┐┌───┐┌───┐ │ │ │T28││T29││T30││T31│ │ │ └───┘└───┘└───┘└───┘ │ │ │ │ 所有线程执行同一条指令 │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ instruction: a b c │ │ │ └──────────────────────────────────────────────────────────┘ │ │ │ │ 但每个线程操作不同的数据 │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ T0: a0 b0 c0 │ │ │ │ T1: a1 b1 c1 │ │ │ │ ... │ │ │ │ T31: a31 b31 c31 │ │ │ └──────────────────────────────────────────────────────────┘ │ └──────────────────────────────────────────────────────────────────┘4.2 Warp的执行过程时间轴 → ┌──────────────────────────────────────────────────────────────────┐ │ │ │ Warp调度器每次选择一个Warp执行一条指令 │ │ │ │ ┌────────────┐ │ │ │ Warp 0 │ ──→ 指令1 (abc) │ │ └────────────┘ │ │ ↓ │ │ ┌────────────┐ │ │ │ Warp 1 │ ──→ 指令1 (abc) │ │ └────────────┘ │ │ ↓ │ │ ┌────────────┐ │ │ │ Warp 2 │ ──→ 指令1 (abc) │ │ └────────────┘ │ │ ↓ │ │ ┌────────────┐ │ │ │ Warp 0 │ ──→ 指令2 (da*e) ← 第一个Warp回来了 │ │ └────────────┘ │ │ │ │ 如果Warp 0在等待数据比如访存 │ │ ┌────────────┐ │ │ │ Warp 0 │ ──→ 访存指令 (load) │ │ └────────────┘ │ │ ↓ (等待数据) │ │ ┌────────────┐ │ │ │ Warp 3 │ ──→ 执行计算 (不用等) ← 零开销切换 │ │ └────────────┘ │ │ ↓ │ │ ┌────────────┐ │ │ │ Warp 4 │ ──→ 执行计算 │ │ └────────────┘ │ │ ↓ │ │ ┌────────────┐ │ │ │ Warp 0 │ ──→ 数据回来了继续执行下一指令 │ │ └────────────┘ │ │ │ └──────────────────────────────────────────────────────────────────┘五、计算核心的进化CUDA Core → Tensor Core5.1 CUDA Core vs Tensor Core 对比图┌──────────────────────────────────────────────────────────────────┐ │ CUDA Core (标量单元) │ │ │ │ 一次做一个运算 │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ a b * c d │ │ │ │ │ │ │ │ b ──┐ │ │ │ │ ├─→ [MUL] ──→ [ADD] ──→ a │ │ │ │ c ──┘ ↑ │ │ │ │ │ │ │ │ │ d ─────────────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ │ │ │ Tensor Core (矩阵单元) │ │ │ │ 一次做一个4×4矩阵乘加 │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ D A × B C │ │ │ │ │ │ │ │ ┌─────────┐ │ │ │ │ │ A (4x4) │ │ │ │ │ └─────────┘ │ │ │ │ × │ │ │ │ ┌─────────┐ │ │ │ │ │ B (4x4) │ │ │ │ │ └─────────┘ │ │ │ │ │ │ │ │ ┌─────────┐ ┌─────────┐ │ │ │ │ │ A×B │ │ C (4x4) │ D (4x4) │ │ │ │ │ (4x4) │ └─────────┘ │ │ │ │ └─────────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ │ │ │ 一次Tensor Core运算 64次CUDA Core运算 │ └──────────────────────────────────────────────────────────────────┘5.2 各代Tensor Core演进┌──────────────────────────────────────────────────────────────────┐ │ Tensor Core 代际演进 │ ├──────────────────────────────────────────────────────────────────┤ │ │ │ Volta (V100, 2017) ──────────────────┐ │ │ ┌──────────────────────────────────┐│ │ │ │ 支持FP16输入FP32累加 ││ │ │ │ 性能125 TFLOPS (FP16) ││ │ │ └──────────────────────────────────┘│ │ │ ↓ │ │ │ Turing (T4, 2018) ─────────────────┐│ │ │ ┌──────────────────────────────────┐││ │ │ │ 新增INT8、INT4量化支持 │││ │ │ │ 性能65 TFLOPS (FP16) │↓│ │ │ │ 130 TOPS (INT8) │ │ │ │ └──────────────────────────────────┘ │ │ │ ↓ │ │ │ Ampere (A100, 2020) ───────────────┐│ │ │ ┌──────────────────────────────────┐││ │ │ │ 新增BF16、TF32、结构化稀疏 │││ │ │ │ 性能312 TFLOPS (FP16) │↓│ │ │ │ 624 TOPS (稀疏) │ │ │ │ └──────────────────────────────────┘ │ │ │ ↓ │ │ │ Hopper (H100, 2022) ───────────────┐│ │ │ ┌──────────────────────────────────┐││ │ │ │ 新增FP8、Transformer引擎 │││ │ │ │ 性能1979 TFLOPS (FP8) │↓│ │ │ │ 3958 TOPS (稀疏) │ │ │ │ └──────────────────────────────────┘ │ │ │ ↓ │ │ │ Blackwell (B200, 2024) ────────────┘ │ │ ┌──────────────────────────────────┐ │ │ │ 新增FP4、第二代Transformer引擎 │ │ │ │ 性能4500 TFLOPS (FP8) │ │ │ │ 9000 TOPS (稀疏) │ │ │ └──────────────────────────────────┘ │ │ │ └──────────────────────────────────────────────────────────────────┘六、内存层次结构6.1 GPU内存层次图速度 (周期) 容量 (每SM) ↓ ↓ 寄存器 ┌─────────┐ ┌─────────┐ 1周期 │ │ 最快 │ ~20KB │ 最小 └─────────┘ └─────────┘ ↓ ↓ 共享内存/L1 ┌─────────┐ ┌─────────┐ ~30周期 │ │ │164KB │ └─────────┘ └─────────┘ ↓ ↓ 只读常量缓存 ┌─────────┐ ┌─────────┐ ~30周期 │ │ │ ~50KB │ └─────────┘ └─────────┘ ↓ ↓ 纹理缓存 ┌─────────┐ ┌─────────┐ ~100周期 │ │ │ ~50KB │ └─────────┘ └─────────┘ ↓ ↓ L2缓存 ┌─────────┐ ┌─────────┐ ~200周期 │ │ │ 40MB │ (全局) └─────────┘ │ (A100) │ ↓ ↓ 全局内存 ┌─────────┐ ┌─────────┐ ~400周期 │ │ 最慢 │40-80GB │ 最大 └─────────┘ └─────────┘ 速度差异寄存器 : 共享内存 : 全局内存 ≈ 1 : 30 : 400七、从硬件到软件的映射7.1 CUDA线程层次与硬件映射┌──────────────────────────────────────────────────────────────────┐ │ CUDA软件概念 → GPU硬件映射 │ ├──────────────────────────────────────────────────────────────────┤ │ │ │ 软件概念层: │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ Grid (网格) │ │ │ │ ┌──────┐ ┌──────┐ ┌──────┐ ┌──────┐ │ │ │ │ │Block0│ │Block1│ │Block2│ │Block3│ ...... │ │ │ │ └──────┘ └──────┘ └──────┘ └──────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ │ ↓ │ │ ↓ (由GigaThread引擎分发) │ │ ↓ │ │ 硬件层: │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ GPU芯片 │ │ │ │ ┌────────┐ ┌────────┐ ┌────────┐ ┌────────┐ │ │ │ │ │ SM0 │ │ SM1 │ │ SM2 │ │ SM3 │ ...... │ │ │ │ └────────┘ └────────┘ └────────┘ └────────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ │ ↓ │ │ ↓ (Block分配到SM) │ │ ↓ │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ SM内部 │ │ │ │ ┌────────────────────────────────────────────────────┐ │ │ │ │ │ Block分配到SM后切分成Warp执行 │ │ │ │ │ │ │ │ │ │ │ │ Warp 0: 线程 0-31 ──→ 调度到CUDA Core执行 │ │ │ │ │ │ Warp 1: 线程32-63 ──→ 调度到CUDA Core执行 │ │ │ │ │ │ Warp 2: 线程64-95 ──→ 调度到CUDA Core执行 │ │ │ │ │ │ Warp 3: 线程96-127 ──→ 调度到CUDA Core执行 │ │ │ │ │ │ ... │ │ │ │ │ └────────────────────────────────────────────────────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ └──────────────────────────────────────────────────────────────────┘7.2 硬件资源限制示意图┌──────────────────────────────────────────────────────────────────┐ │ SM资源限制示意图 │ ├──────────────────────────────────────────────────────────────────┤ │ │ │ 一个SM最多能同时容纳 │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ 2048个线程 64个Warp × 32线程 │ │ │ └──────────────────────────────────────────────────────────┘ │ │ ↓ │ │ 但实际受限于 │ │ │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ 寄存器限制 │ │ │ │ ┌────────────────────────────────────────────────────┐ │ │ │ │ │ 寄存器总数: 65536 │ │ │ │ │ │ 每个线程用32寄存器 → 2048线程 ✅ │ │ │ │ │ │ 每个线程用64寄存器 → 1024线程 ❌ │ │ │ │ │ │ 每个线程用128寄存器 → 512线程 ❌ │ │ │ │ │ └────────────────────────────────────────────────────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ │ │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ 共享内存限制 │ │ │ │ ┌────────────────────────────────────────────────────┐ │ │ │ │ │ 共享内存总数: 164KB │ │ │ │ │ │ 每个Block用16KB → 10个Block (164÷16≈10) ❌ │ │ │ │ │ │ 每个Block用32KB → 5个Block (164÷32≈5) ❌ │ │ │ │ │ │ 每个Block用48KB → 3个Block (164÷48≈3) ❌ │ │ │ │ │ └────────────────────────────────────────────────────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ │ │ │ ┌──────────────────────────────────────────────────────────┐ │ │ │ 线程块限制 │ │ │ │ ┌────────────────────────────────────────────────────┐ │ │ │ │ │ 最大Block数: 32 │ │ │ │ │ │ 即使寄存器/共享内存够用也不能超过32个Block ❌ │ │ │ │ │ └────────────────────────────────────────────────────┘ │ │ │ └──────────────────────────────────────────────────────────┘ │ │ │ │ 实际活跃Warp数 min(理论最大, 寄存器限制, 共享内存限制, Block限制) │ │ Occupancy 实际活跃Warp数 / 64 │ └──────────────────────────────────────────────────────────────────┘八、NVIDIA核心性能参数全景图下面的表格汇总了从Volta到Blackwell架构的关键型号在编程和模型选型时最关注的硬指标型号架构FP16 算力 (TFLOPS)BF16 算力 (TFLOPS)FP8 算力 (TFLOPS)INT8 算力 (TOPS)FP64 算力 (TFLOPS)显存带宽显存容量关键互联技术V100Volta125不原生支持不原生支持不原生支持7.8900 GB/s32GB HBM2NVLink 2.0A100Ampere312312不原生支持62419.52.0 TB/s (HBM2e)80GB HBM2eNVLink 3.0 (600 GB/s)A800 (中国特供版)Ampere312312不原生支持62419.52.0 TB/s (HBM2e)80GB HBM2eNVLink 带宽受限 (400 GB/s)H100Hopper1,9791,9793,9583,958673.35 TB/s (HBM3)80GB HBM3NVLink 4.0 (900 GB/s)H800 (中国特供版)Hopper1,9791,9793,9583,958673.35 TB/s (HBM3)80GB HBM3NVLink 带宽受限 (400 GB/s)H200Hopper~1,979~1,979~3,958~3,958674.8 TB/s (HBM3e)141GB HBM3eNVLink 4.0 (900 GB/s)L40SAda Lovelace7337331,4661,46691.6(FP32)864 GB/s (GDDR6)48GB GDDR6不支持NVLinkB100Blackwell待发布待发布~7,900 (预估)待发布待发布~4.8 TB/s (HBM3e)192GB HBM3eNVLink 5.0 (1.8 TB/s)RTX 5090Blackwell83(FP32)83支持支持1.3 (典型值)~1 TB/s (GDDR7)32GB GDDR7不支持RTX PRO 4500Blackwell支持支持支持 (含FP4)支持1.4 (典型值)896 GB/s (GDDR7 with ECC)32GB GDDR7 with ECC不支持H100 NVL (94G)Hopper1,9791,9793,9583,958673.9 TB/s (HBM3)94GB HBM3(单卡)NVLink 4.0 编程调优的关键洞察这些参数对于编程实践意味着什么精度选择直接决定性能上限对于大模型训练H100及后续架构对FP8的原生支持是关键。它能在保持模型精度的同时大幅提升计算吞吐量并降低显存占用是降低万亿参数模型训练成本的核心技术 。BF16在A100及之后架构上成为主流因其动态范围与FP32相同简化了混合精度训练的难度 。如果你的工作负载涉及科学计算如分子动力学、气候模拟FP64性能是刚需。H100的FP64性能是A100的3倍以上而V100至今仍在这一领域有应用 。内存带宽与容量大模型的“命门”H200的4.8TB/s带宽和141GB容量是一个分水岭。它意味着像Llama 3 70B这样的模型可以单卡部署无需复杂的模型并行策略从而显著降低推理延迟并简化工程实现 。对于编程而言这意味着可以尝试更长的序列长度和更大的batch size。H100 NVL (94G)的双卡配置为参数量超过80B的模型提供了一种高带宽、大容量的解决方案尤其适合推理场景 。互联技术决定多卡扩展效率在进行多卡分布式训练时NVLink带宽至关重要。H100的900GB/s NVLink 能保证梯度同步等通信操作几乎不成为瓶颈实现近线性的加速比。特别留意中国特供版A800/H800其NVLink带宽被显著限制 。如果你的训练任务对卡间通信极为敏感如使用GSPMD等并行策略这个差异会对编程和性能优化策略产生重要影响。架构特性针对性优化Transformer Engine (H100)编程时可以利用其自动在FP8和FP16间切换的能力简化了低精度训练的代码复杂度 。L40S它拥有RT核心在AI推理的同时具备强大的3D图形渲染能力 。如果你的应用是AI图形学如数字人、Omniverse这是编程时需要重点利用的异构计算特性。RTX PRO 4500作为工作站显卡它支持ECC显存纠错这对需要长时间运行、数据零差错的专业计算和科学工作流至关重要 。九、动手实践9.1 查询你的GPU信息运行以下代码查看你的GPU硬件参数#includecuda_runtime.h#includestdio.hintmain(){intdeviceCount;cudaGetDeviceCount(deviceCount);for(inti0;ideviceCount;i){cudaDeviceProp prop;cudaGetDeviceProperties(prop,i);printf(GPU %d: %s\n,i,prop.name);printf( 计算能力: %d.%d\n,prop.major,prop.minor);printf( SM数量: %d\n,prop.multiProcessorCount);printf( 每个SM最大线程数: %d\n,prop.maxThreadsPerMultiProcessor);printf( 每个SM最大Warp数: %d\n,prop.maxThreadsPerMultiProcessor/32);printf( 最大线程块大小: %d\n,prop.maxThreadsPerBlock);printf( 寄存器数量/SM: %d\n,prop.regsPerMultiprocessor);printf( 共享内存大小/SM: %.2f KB\n,prop.sharedMemPerMultiprocessor/1024.0);printf( 全局内存大小: %.2f GB\n,prop.totalGlobalMem/(1024.0*1024*1024));printf( L2缓存大小: %d KB\n,prop.l2CacheSize/1024);printf(----------------------------------------\n);}return0;}9.2 编译运行nvcc device_query.cu-odevice_query ./device_query十、本节总结10.1 核心要点回顾GPU是吞吐优先的设计几千个简单核心通过大量线程隐藏延迟SM是核心计算单元每个SM有自己的CUDA Core、Tensor Core、寄存器、共享内存Warp是执行最小单位32个线程一起执行同一条指令零开销调度通过切换Warp隐藏访存延迟Tensor Core是AI加速核心一次完成64次乘加运算Occupancy决定性能受寄存器、共享内存限制10.2 下节预告下一节我们将深入GPU内存体系搞懂全局内存为什么慢共享内存怎么用才能快寄存器溢出会怎样如何通过内存优化让性能翻倍十一、参考资料NVIDIA ADAADA Architectute NvidiaNVIDIA Hopper Architectute NvidiaNVIDIA Ampere GPU ArchitectureNVIDIA GPU TuringNVIDIA V100 数据中心GPU (官网)NVIDIA A100 GPU 为现代数据中心提供强大动能 (官网)NVIDIA A100 GPUs Power the Modern Data Center (官网英文)NVIDIA H100 Tensor Core GPU (官网英文)NVIDIA H200 GPU (官网英文)NVIDIA Enterprise Reference Architecture: Appendix B (NVIDIA官方文档)NVIDIA Enterprise Reference Architecture: Appendix A (NVIDIA官方文档)NVIDIA GeForce RTX 5090 显卡 (官网)十二、面试真题2024-2026Q1解释GPU的SIMT模型和CPU的SIMD有什么区别考察点对两种并行模型的理解参考答案SIMD单指令多数据一条指令同时操作多个数据所有单元步调一致如CPU的AVX指令。数据必须连续存储处理分支困难。SIMT单指令多线程多个线程执行同一条指令但每个线程可以访问不同地址、可以有独立分支虽然分支会序列化。SIMT更灵活适合图形学和AI等不规则并行任务这也是GPU适合AI的原因。Q2什么是WarpWarp的大小为什么是32考察点对硬件细节的了解参考答案Warp是GPU调度的最小单位32个线程组成一个warp。32是设计权衡的结果太小则调度开销占比大太大则分支开销大、资源浪费。从Tesla架构开始就定为32一直延续至今保证了软件生态的兼容性。Q3什么是Occupancy如何计算和优化考察点性能优化基础参考答案Occupancy 活跃warp数 / 最大支持warp数通常64。计算方法受限于寄存器、共享内存、线程块大小。优化方法减少每个线程使用的寄存器-maxrregcount编译选项减少每个Block使用的共享内存调整Block大小128-256通常较好使用__launch_bounds__提示编译器Q4Tensor Core和CUDA Core有什么区别分别在什么场景使用考察点对现代GPU架构的理解参考答案CUDA Core通用标量单元适合控制流复杂、非规则计算。Tensor Core专用矩阵乘加单元一次完成4×4矩阵乘加64次运算。使用场景CUDA Core自定义算子、元素级操作、复杂逻辑Tensor Core矩阵乘法、卷积、AI训练/推理开发中一般用cuBLAS、cuDNN自动调用Tensor Core手写需用wmma命名空间API。Q5SM上同时能跑多少个线程受什么限制考察点对硬件资源约束的理解参考答案以A100为例理论上限2048线程64 warp × 32线程。实际受三重限制寄存器限制65536寄存器 ÷ 每线程寄存器数共享内存限制164KB ÷ 每Block共享内存 × 每Block线程数线程块限制最多32个Block同时驻留最终取三者最小值。思考题如果你的核函数每个线程用64个寄存器每个Block有256个线程每个Block用32KB共享内存。在A100上一个SM最多能同时跑多少个线程提示A100每个SM有65536个寄存器、164KB共享内存、最多64个warp、最多32个Block。计算。Tensor Core专用矩阵乘加单元一次完成4×4矩阵乘加64次运算。使用场景CUDA Core自定义算子、元素级操作、复杂逻辑Tensor Core矩阵乘法、卷积、AI训练/推理开发中一般用cuBLAS、cuDNN自动调用Tensor Core手写需用wmma命名空间API。Q5SM上同时能跑多少个线程受什么限制考察点对硬件资源约束的理解参考答案以A100为例理论上限2048线程64 warp × 32线程。实际受三重限制寄存器限制65536寄存器 ÷ 每线程寄存器数共享内存限制164KB ÷ 每Block共享内存 × 每Block线程数线程块限制最多32个Block同时驻留最终取三者最小值。思考题如果你的核函数每个线程用64个寄存器每个Block有256个线程每个Block用32KB共享内存。在A100上一个SM最多能同时跑多少个线程提示A100每个SM有65536个寄存器、164KB共享内存、最多64个warp、最多32个Block。欢迎在评论区留下你的答案
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2424111.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!