计算着色器中缓冲区索引的理解

Question

计算着色器脚本：

#pragma kernel CSMain
RWStructuredBuffer<int> result;
 
[numthreads(8, 8, 8)]
void CSMain(int id : SV_DispatchThreadID )
{
    result[id] = id;
}

c# 脚本：

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class ForTesting : MonoBehaviour
{
    public ComputeShader compute;
 
    // Use this for initialization
    void Start()
    {
        ComputeBuffer resultBuffer = new ComputeBuffer(8*8*8, sizeof(int));
     
        var kernel = compute.FindKernel("CSMain");
        compute.SetBuffer(kernel, "result", resultBuffer);
        compute.Dispatch(kernel,1,1,1);
 
        int[] result = new int[8 * 8 * 8];
        resultBuffer.GetData(result);
   
        resultBuffer.Release();
    }
}

这些是测试结果中的值的代码。 我不明白 result[id] = id; 线。

没错， result[id] 。 使用索引访问缓冲区。

在google的时候，我看到一个帖子说着色器function执行的线程数一样多。

那么在上述情况下，导致 numthreads(8, 8, 8) 和 Dispatch(kernel, 1, 1, 1)，CSMain function 运行 8 * 8 * 8 * 1 * 1 * 1 次吗？

Answer 1

首先result[id] = id; 不是类型安全的。 id具有三个int组件，但result的索引和result的值只是一个 int。 所以真正发生的只是第一个组件正在被使用，即result[id.x] = id.x;

您可以通过id.y和id.z访问第二个和第三个元素。

---

您可以将计算着色器视为一个 for 循环

void kernel(int[3] id)
{
    ...
}

List<int[3]> ids;

for (var id : ids)
    kernel(id);

除了对 kernel 的每个调用都是并行执行的，而不是按顺序执行的。

---

您应该阅读此处的文档以了解List<int[3]> ids; 即计算SV_DispatchThreadID

• https://docs.microsoft.com/en-us/windows/win32/direct3dhlsl/sm5-attributes-numthreads
• https://docs.microsoft.com/en-us/windows/win32/api/d3d11/nf-d3d11-id3d11devicecontext-dispatch

在您的具体情况下：

var iota = Enumerable.Range(0, 8);
List<int[3]> ids = cartesian_product(iota, iota, iota);

其中 ids 的大小/长度为8*8*8 ；