后端岗位面试题更新 2026-08-05

请描述如何优化基于 CUDA 的 GEMM(矩阵乘法)实现,并说明其中 shared memory 的大小应如何确定。

快手后端开发互联网/IT性能优化系统设计技术原理CUDA

考察说明

考察 CUDA 编程中 GEMM 优化思路及 shared memory 分配的计算方法

回答思路

  1. 能阐述分块策略与 shared memory 的利用率
  2. 能说明 shared memory 大小受硬件限制(如 48KB/block)与块大小、数据类型的影响
  3. 能结合 bank conflict、寄存器占用等细节说明优化
  4. 能解释如何平衡 shared memory 大小与 occupancy