Metal计算着色器：如何访问更大纹理区域实现长滤镜运动模糊？

首先纠正你的误解：线程组大小不会限制单个线程可访问的纹理范围，它仅影响GPU计算任务的调度效率，调整它无法解决你当前的模糊效果未增强的问题。你的问题根源在于滤镜长度的计算逻辑错误，导致采样次数与权重不匹配，同时存在浮点数精度隐患。

1. 核心问题：滤镜范围计算错误 #

当你将filter_len设为30时，原代码的循环次数实际是31次（而非预期的30次），且每个采样点的权重是1/30，最终总和会超过1，导致图像过亮且模糊效果不符合预期：

• 原代码中，filter_len=30时，floor(30/2)=15，lower_bound=-15，upper_bound=15+1=16，循环从i=-15到i=15（共31次）
• 31次采样的权重总和为31/30≈1.033，破坏了颜色值的归一化，同时采样次数与滤镜长度不匹配

2. 修复步骤 #

步骤1：改用整数类型存储滤镜长度 #

将filter_len从half改为int，避免浮点数精度问题：

const int filter_len = 30; // 直接用整数定义滤镜长度

步骤2：修正采样范围计算逻辑 #

确保循环次数严格等于滤镜长度，同时保证采样范围对称：

int half_len = filter_len / 2;
int lower_bound = -half_len;
int upper_bound = filter_len - half_len; // 确保循环次数等于filter_len

• 当filter_len=30时：half_len=15，lower_bound=-15，upper_bound=15，循环从i=-15到i=14（共30次）
• 当filter_len=15时：half_len=7，lower_bound=-7，upper_bound=8，循环从i=-7到i=7（共15次），与原逻辑一致，兼容奇数长度

步骤3：优化采样坐标（可选） #

由于你使用coord::pixel类型的采样器，纹理坐标的像素中心是(x+0.5, y+0.5)，而globalId是像素的索引值（左上角坐标），可以调整采样坐标以确保采样的是像素中心：

float2 c = static_cast<float2>(globalId) + float2(0.5, 0.5);

这会让采样更精准，但不是必须的，因为你使用整数偏移采样，最终结果差异不大。

3. 修复后的完整代码 #

constexpr sampler s(coord::pixel, address::clamp_to_edge);

kernel void motion_blur(texture2d<half, access::sample> gray_source [[ texture(0) ]],
                        texture2d<half, access::write> b_hor [[ texture(1) ]],
                        texture2d<half, access::write> b_ver [[ texture(2) ]],
                        uint2 globalId [[ thread_position_in_grid ]]) {
    float2 c = static_cast<float2>(globalId) + float2(0.5, 0.5); // 可选：采样像素中心
    
    // 改用整数定义滤镜长度，修正范围计算
    const int filter_len = 30;
    int half_len = filter_len / 2;
    int lower_bound = -half_len;
    int upper_bound = filter_len - half_len;
    
    half g_x = 0;
    half g_y = 0;
    half weight = 1.0 / filter_len; // 提前计算权重，避免重复除法
    for (int i = lower_bound; i < upper_bound; i++) {
        half xGray = gray_source.sample(s, c + float2(i, 0)).x * weight;
        g_x += xGray;
        half yGray = gray_source.sample(s, c + float2(0, i)).x * weight;
        g_y += yGray;
    }
    b_hor.write(g_x, globalId);
    b_ver.write(g_y, globalId);
}

4. 关于线程组大小的补充 #

线程组大小建议保持16x16，这是苹果GPU的最优调度单元之一（匹配GPU的 warp 大小）。如果改为32x32，部分GPU可能会因为线程组线程数（1024）接近上限而出现性能下降，但不会影响纹理访问范围。

内容的提问来源于stack exchange，提问作者whlteXbread