Unity中不回传主机合并多个ComputeBuffer的实现方案

方案1：直接传递多个Buffer到目标Shader #

Unity的C# API支持给Shader中声明的固定长度Buffer数组逐个绑定，无需合并数据。

步骤： #

1. 在目标ComputeShader中声明固定长度的Buffer数组（根据需求设一个足够大的上限，比如64），同时传入Buffer数量：

// 目标ComputeShader代码片段
StructuredBuffer<float> inputBuffers[64];
int bufferCount; // 实际传入的Buffer数量

2. C#端循环绑定每个Buffer，并设置数量参数：

// C#代码片段
int kernelIndex = targetCS.FindKernel("YourKernelName");
int bufferCount = inputBuffers.Length;

// 逐个绑定数组中的Buffer
for (int i = 0; i < bufferCount; i++)
{
    targetCS.SetComputeBufferParam(kernelIndex, "_InputBuffers", i, inputBuffers[i]);
}
targetCS.SetInt("_BufferCount", bufferCount);

// 执行Dispatch
targetCS.Dispatch(kernelIndex, threadGroupsX, 1, 1);

3. 在Shader内核中遍历Buffer数组处理数据：

[numthreads(64,1,1)]
void YourKernelName (uint3 id : SV_DispatchThreadID)
{
    // 示例：遍历每个Buffer的对应元素
    for (int i = 0; i < bufferCount; i++)
    {
        float value = inputBuffers[i][id.x];
        // 处理逻辑...
    }
}

方案2：GPU端合并为单个扁平Buffer #

如果必须用单个Buffer，写一个专门的合并ComputeShader，在GPU端完成数据拼接，全程无需回传主机。

步骤： #

1. 创建合并用的ComputeShader：

#pragma kernel MergeBuffers

StructuredBuffer<float> inputBuffers[64];
RWStructuredBuffer<float> outputBuffer;
int maskSize; // 单个掩码的元素数量
int bufferCount; // 输入Buffer总数

[numthreads(64,1,1)]
void MergeBuffers (uint3 id : SV_DispatchThreadID)
{
    // 计算当前线程对应的Buffer索引和元素索引
    int bufferIndex = id.x / maskSize;
    int elementIndex = id.x % maskSize;

    // 边界检查
    if (bufferIndex >= bufferCount || elementIndex >= maskSize)
        return;

    // 拷贝元素到输出Buffer
    outputBuffer[id.x] = inputBuffers[bufferIndex][elementIndex];
}

2. C#端准备输出Buffer并执行合并：

// C#代码片段
int maskSize = 1024; // 单个掩码的元素数
int bufferCount = inputBuffers.Length;
int totalElements = maskSize * bufferCount;

// 创建输出Buffer，注意Stride和输入Buffer一致（float类型为4字节）
ComputeBuffer mergedBuffer = new ComputeBuffer(totalElements, sizeof(float));

int mergeKernel = mergeCS.FindKernel("MergeBuffers");
// 绑定输入Buffer数组
for (int i = 0; i < bufferCount; i++)
{
    mergeCS.SetComputeBufferParam(mergeKernel, "_InputBuffers", i, inputBuffers[i]);
}
mergeCS.SetComputeBufferParam(mergeKernel, "_OutputBuffer", mergedBuffer);
mergeCS.SetInt("_MaskSize", maskSize);
mergeCS.SetInt("_BufferCount", bufferCount);

// 计算Dispatch线程组数
int threadGroups = Mathf.CeilToInt(totalElements / 64f);
mergeCS.Dispatch(mergeKernel, threadGroups, 1, 1);

// 此时mergedBuffer就是合并后的结果，可直接传给目标ComputeShader使用

方案3：动态合并（AppendBuffer） #

如果各掩码长度不一致，用AppendStructuredBuffer实现动态拼接，无需提前计算总长度：

合并Shader代码： #

#pragma kernel MergeWithAppend

StructuredBuffer<float> inputBuffers[64];
AppendStructuredBuffer<float> outputBuffer;
int maskSize;
int bufferCount;

[numthreads(64,1,1)]
void MergeWithAppend (uint3 id : SV_DispatchThreadID)
{
    int bufferIndex = id.x / maskSize;
    int elementIndex = id.x % maskSize;

    if (bufferIndex >= bufferCount || elementIndex >= maskSize)
        return;

    outputBuffer.Append(inputBuffers[bufferIndex][elementIndex]);
}

C#端注意事项： #

创建AppendStructuredBuffer时无需指定总长度，合并完成后如果需要重置，可调用mergedBuffer.SetCounterValue(0);，同时需要配合ComputeBuffer.CounterType.Append参数创建Buffer：

ComputeBuffer mergedBuffer = new ComputeBuffer(0, sizeof(float), ComputeBufferType.Append);

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