Swift环境下如何对2DArray类型MTLTexture切片并传入MPSImageHistogram？

核心实现思路 #

MPSImageHistogram 原生支持直接处理2DArray纹理的单个切片，无需调用makeTextureView()创建额外纹理视图，仅需在编码命令时显式指定目标切片索引即可，全程无额外内存或对象创建开销。

具体实现步骤 #

• 首先初始化MPSImageHistogram实例，按业务需求配置直方图参数：

import MetalPerformanceShaders

// 示例配置：256个统计桶，忽略Alpha通道，像素值范围0~1
var histogramInfo = MPSImageHistogramInfo(
    numberOfHistogramEntries: 256,
    histogramForAlpha: false,
    minPixelValue: vector_float4(0, 0, 0, 0),
    maxPixelValue: vector_float4(1, 1, 1, 1)
)
let histogramKernel = MPSImageHistogram(device: metalDevice, histogramInfo: &histogramInfo)

• 编码计算命令时直接传入原始2DArray纹理，通过sourceSlice参数指定要计算的切片索引：

guard let commandBuffer = commandQueue.makeCommandBuffer() else { return }

// sourceSlice传入目标切片的索引（从0开始计数），sourceLevel填0表示使用基础mip层级
histogramKernel.encode(
    to: commandBuffer,
    sourceTexture: target2DArrayTexture,
    sourceSlice: requiredSliceIndex,
    sourceLevel: 0,
    destinationHistogram: outputHistogramBuffer,
    destinationHistogramOffset: 0
)

commandBuffer.commit()
// 异步读取结果或等待计算完成
commandBuffer.waitUntilCompleted()

注意事项 #

该方案是Metal官方推荐的单切片处理方式，相比makeTextureView()仅传递切片索引参数，完全避免了纹理视图对象的创建、状态校验开销，性能表现最优。

• 传入的sourceSlice必须小于原始纹理的arrayLength属性，否则会触发Metal运行时校验错误
• 此参数配置方式对所有支持切片指定的MPS算子通用，不限于直方图计算场景
• 多切片批量计算时可复用同一个MPSImageHistogram实例，仅循环传入不同sourceSlice即可

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