计算着色器

什么是计算着色器

计算着色器（Compute Shader）是一种可编程着色器，允许开发者利用GPU的大规模并行计算能力处理通用任务，而不仅限于图形渲染。它运行在独立的计算管线上，可以读写缓冲区和纹理，适合处理大量数据并行的问题。

核心概念

线程与工作组

计算着色器以线程（Thread）为基本执行单元，多个线程组成工作组（Work Group）。工作组内的线程可以通过共享内存（Shared Memory）进行通信和同步。所有工作组构成一个全局的调度空间，通常用三维索引（x, y, z）标识。

着色器存储对象

计算着色器可以访问多种资源：

• **结构化缓冲区（Structured Buffer）**：存储自定义数据结构。
• **原始缓冲区（Raw Buffer）**：存储原始字节数据。
• **读写纹理（RWTexture）**：可直接读写纹理，无需渲染管线。
• **常量缓冲区（Constant Buffer）**：传递常量参数。

同步机制

工作组内线程可通过屏障（Barrier）同步执行，确保共享内存的一致性。不同工作组之间无法直接同步，通常通过多次调度或原子操作协调。

与图形管线的区别

计算着色器独立于图形管线，没有固定的输入装配、光栅化等阶段。它直接由应用程序调度执行，无需绑定顶点缓冲区、渲染目标等。因此，它更适合非图形任务，但也可以用于优化图形相关的计算（如剔除、LOD生成）。

应用场景

后处理与图像处理

计算着色器常用于模糊、色调映射、直方图计算等后处理效果，比基于像素着色器的全屏四边形更灵活高效。

粒子系统

模拟海量粒子的运动、碰撞，存储粒子状态在缓冲区，计算着色器更新位置和速度。

物理模拟

例如布料、流体、软体等，利用并行性加速数值计算。

剔除与场景管理

在计算着色器中进行视锥体裁剪、遮挡剔除，减少渲染批次。

通用计算

如排序、矩阵运算、神经网络推理等，类似GPGPU。

编程模型与API

在DirectX 11+中，使用HLSL编写计算着色器，通过`Dispatch`调用；在OpenGL 4.3+中使用GLSL，通过`glDispatchCompute`调用。着色器代码中需要声明`numthreads`指定工作组大小。典型工作流程：

1. 创建计算着色器并编译。
2. 绑定输入/输出资源（UAV、SRV等）。
3. 调用`Dispatch`指定工作组数量。
4. GPU执行着色器，结果写回资源。

优势与挑战

**优势**：

• 高度并行，适合数据并行任务。
• 绕过图形管线，减少固定功能开销。
• 共享内存可加速工作组内通信。

**挑战**：

• 需要理解GPU架构，合理安排工作组大小和内存访问模式。
• 调试较困难，需要工具支持（如NVIDIA Nsight、RenderDoc）。
• 不同硬件之间性能差异大。

实践建议

• 尽量合并内存访问，避免随机读写。
• 合理选择工作组大小，通常为64或128线程，但需测试。
• 利用共享内存减少全局内存访问。
• 用多个小Dispatch代替一个大Dispatch，便于负载均衡。