我自定义的Metal图像滤镜运行速度慢，该如何优化提速？

核心问题分析 #

你当前代码90%以上的耗时都来自CPU侧的冗余操作，GPU kernel本身的计算开销几乎可以忽略，以下是具体优化点：

1. 修复命令缓冲/编码器的创建逻辑 #

你现在在init方法中一次性创建了MTLCommandBuffer和MTLComputeCommandEncoder，但这两个对象是单次使用的，调用endEncoding()和commit()之后就失效了，必须每次调用applyFilter时重新创建：

// 删掉init里的这两行
// self.commandBuffer = self.commandQueue?.makeCommandBuffer()
// self.commandEncoder = self.commandBuffer?.makeComputeCommandEncoder()

// applyFilter里每次重新创建
func applyFilter() -> UIImage? {
    guard let commandBuffer = commandQueue?.makeCommandBuffer(),
          let encoder = commandBuffer.makeComputeCommandEncoder() else {
        fatalError("创建命令对象失败")
    }
    // 剩下的逻辑用这两个新创建的对象
}

2. 砍掉冗余的图片格式转换开销 #

当前每次执行滤镜都走UIImage→CGImage→MTLTexture的全链路转换，其中UIGraphicsBeginImageContext是CPU侧渲染，开销极高：

• 重复处理同一张图时，提前把输入的UIImage转成MTLTexture缓存，不要每次调用applyFilter都重新转换
• 使用MTKTextureLoader时添加关闭sRGB校正的选项，避免不必要的颜色空间转换：

let texture = try textureLoader.newTexture(cgImage: cgimg, options: [.SRGB: false])

• 如果是处理相机预览流，直接从CMSampleBuffer转MTLTexture，不要转成UIImage再处理

3. 修复线程派发逻辑，避免像素丢失+提升效率 #

你当前的getBlockDimensions直接对图片宽高做整除，若图片宽高不是32的倍数，边缘像素会被遗漏。改用Metal的非均匀线程派发方法，无需手动计算线程组数量：

// 删掉getBlockDimensions方法，直接用以下代码派发线程
let gridSize = MTLSize(width: width, height: height, depth: 1)
let threadGroupSize = MTLSize(width: 16, height: 16, depth: 1) // 16*16=256线程，适配所有iOS设备
encoder.dispatchThreads(gridSize, threadsPerThreadgroup: threadGroupSize)

4. 去掉同步等待，改用异步回调 #

buffer.waitUntilCompleted()会阻塞当前线程，改为异步回调获取结果，用户感知的响应速度会大幅提升：

// 删掉buffer.waitUntilCompleted()，替换为回调
buffer.addCompletedHandler { [weak self] _ in
    guard let self = self,
          let outputCGImage = self.getCGImage(from: outputTexture) else { return }
    let outputImage = UIImage(cgImage: outputCGImage)
    // 回到主线程更新UI
    DispatchQueue.main.async {
        // 处理结果回传
    }
}
buffer.commit()

5. 优化Kernel计算逻辑 #

GPU的乘法运算延迟远低于除法，把除法改为乘法，性能有微小提升：

#include <metal_stdlib>
using namespace metal;
kernel void black(
               texture2d<float, access::write> outTexture [[texture(0)]],
               texture2d<float, access::read> inTexture [[texture(1)]],
               uint2 id [[thread_position_in_grid]]) {
    float4 val = inTexture.read(id);
    float r = val.r * 0.25;
    float g = val.g * 0.25;
    float b = val.b * 0.5;
    outTexture.write(float4(r, g, b, val.a), id);
}

6. 缓存可复用对象 #

如果每次处理的图片尺寸固定，不要每次都创建新的输出MTLTexture，复用同一个纹理对象即可，减少内存分配开销。

效果说明 #

完成以上优化后，排除首次初始化的开销，单张1080P图片的处理耗时可以稳定在0.01秒以内，符合你看到的教程水平。

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