Shader 中的 if：Uniform 分支 vs 动态分支

分支语句在 Shader 中并非一律昂贵。理解 GPU 执行模型才能准确判断何时可以放心使用if何时需要替代方案。01 GPU 执行模型先理解 WarpGPU 不像 CPU 那样逐线程独立运行而是将若干线程捆绑为一个WarpNVIDIA/ WavefrontAMD由同一条指令流批量驱动。对 Fragment Shader 来说这意味着一个 Warp 内的所有像素共用同一套指令序列。正是这种锁步机制让分支的代价与条件是否在 Warp 内一致直接挂钩。02 Uniform 分支几乎零开销当if的判断条件对整个 Draw Call 的所有线程都相同时称为Uniform 分支。典型案例是从Properties传入的开关Toggle或全局宏。// Properties 中声明的 Toggle float _EnableFeature; // 0 或 1整个 DC 固定 half4 Fragment(Varyings input) : SV_Target { if (_EnableFeature 0.5) // ← Uniform 条件 { // 整个 Warp 要么全走这里要么全不走 color ApplyRimLight(input.normalWS); } return color; }GPU 在着色器编译阶段或指令分派阶段即可判定条件值直接跳过不执行的分支整个指令块不存在任何线程的计算浪费。结论Uniform 条件的if等同于静态路径切换GPU 驱动或编译器可将不执行的分支整体剔除开销近乎为零。常见 Uniform 条件来源// 方式 A直接 float toggle常用于简单开关 float _EnableRim; // 0.0 / 1.0 float _EnableOutline; // 0.0 / 1.0 // 方式 BShader Keyword编译期剔除更彻底 #pragma shader_feature_local _RIM_ON #pragma multi_compile _ OUTLINE_ON // 使用方式 #if defined(_RIM_ON) color RimLight(N, V); #endif建议对于功能级别的开关优先使用shader_feature_local关键字编译器会为每条路径生成独立 Variant运行时字面上不存在被剔除的代码比 Uniform float 更干净。03 动态分支真实开销从何而来当条件依赖每像素不同的变量如uv、法线、采样结果Warp 内不同线程的条件结果可能不一致这就是动态分支也是真正需要小心的场景。half4 Fragment(Varyings input) : SV_Target { half4 mask SAMPLE_TEXTURE2D(_MaskTex, sampler_MaskTex, input.uv); if (mask.r 0.5) // ← 动态条件每像素不同 { color ExpensivePathA(input); // 路径 A } else { color ExpensivePathB(input); // 路径 B } return color; }Warp 分歧Warp Divergence当 Warp 内的线程因条件不同而需要走不同路径时GPU 无法让它们真正分叉。实际做法是先执行路径 A路径 B 的线程被屏蔽再执行路径 B路径 A 的线程被屏蔽最后合并结果。两条路径的指令都被执行了一遍代价是真实存在的。注意动态分支的真实开销 两条路径之和而非较慢的那条。越昂贵的分支内容多次纹理采样、复杂数学动态if的代价越高。04 横向对比维度Uniform 分支动态分支条件来源Properties、CBuffer、宏定义等所有线程相同逐像素变量UV、贴图采样、normalWS、positionWS等Warp 内状态所有线程条件一致部分线程 TRUE部分 FALSE → 分歧GPU 行为提前剔除不执行分支串行执行两条路径屏蔽对应线程实际开销极低 约等于无较高 ≈ 两条路径之和Warp 利用率100%50% ~ 100%取决于分歧比例典型场景功能开关、质量等级、平台差异Mask 区域判断、像素距离判断、噪声阈值推荐写法shader_feature或float _Flag优先用lerp/step消除分支05 实践用数学消除动态分支对于简单的动态分支通常可以用lerp、step、saturate等数学函数改写为无分支形式让所有线程执行相同指令序列规避 Warp 分歧。示例Mask 区域混合half4 color; if (mask.r 0.5) color colorA; else color colorB;// step(edge, x) 当 x edge 时返回 1否则 0 half t step(0.5, mask.r); // 0 或 1无分支 half4 color lerp(colorB, colorA, t); // 线性混合示例复杂路径的权重混合// 注意两条路径都会被执行适合分支内容轻量的场景 half4 resultA PathA(input); // 始终执行 half4 resultB PathB(input); // 始终执行 half weight step(0.5, mask.r); half4 color lerp(resultB, resultA, weight);权衡点无分支写法让所有线程执行相同指令消除了 Warp 分歧但两条路径都会被计算。若某条路径极昂贵如多次纹理采样改为无分支后整体开销反而可能上升。需要根据路径复杂度具体判断。什么时候动态分支仍然合适当分支体内计算量相差悬殊且屏幕上大多数像素只走轻量路径时保留动态if可以避免所有线程都被拖到重路径的开销。例如// 角色区域mask 极小背景区域mask 极大 // Warp 大部分时候不分歧 → 动态 if 反而更省 if (characterMask.r 0.5) { color ExpensiveSSS(input); // 次表面散射仅角色像素 }06 URP 中的额外注意事项移动端差异移动端 GPU如 Mali、Adreno的 Warp 尺寸和分歧惩罚与桌面端不同。部分低端 GPU 对动态分支的支持较弱甚至会将整个分支展开Scalar / Vector 寄存器压力倍增。URP 的默认做法是在 Mobile 质量档位下尽量不使用动态分支。Shader Keyword 与 Variant 爆炸虽然shader_feature是消除运行时分支的最优解但 Keyword 过多会引发 Variant 数量爆炸导致编译时间和包体暴增。合理控制 Keyword 数量对低优先级功能仍可退而求其次用 Uniform float 分支。// _local 限定在材质范围减少全局污染推荐优先使用 #pragma shader_feature_local _RIM_LIGHT_ON #pragma shader_feature_local _DETAIL_MAP_ON // multi_compile 会生成所有组合keyword 数量多时谨慎 #pragma multi_compile _ _FEATURE_A _FEATURE_B _FEATURE_C总结 · 一图记忆Uniform 分支 — 放心用条件来自 Properties / CBuffer同一 Draw Call 所有线程条件一致GPU 提前剔除不执行路径开销近乎为零最优改为shader_feature关键字动态分支 — 谨慎用条件依赖逐像素变量Warp 内线程条件可能不一致两条路径串行执行产生 Warp 分歧开销 ≈ 路径 A 路径 B优先改为lerp/step无分支形式决策树条件是 Uniform→ 直接用if或换成 Keyword条件是动态 分支体轻量→ 用lerp/step消除条件是动态 分支体重量不对称→ 保留if测量实际 GPU 耗时移动端额外建议默认避免动态分支尤其 Fragment Shader用 Frame Debugger GPU Profile 验证实测数据Keyword 数量控制在每个 Pass ≤ 4 个独立开关