Unity中利用Compute Shader做GPU加速计算

零度冰山

已于 2023-07-24 10:23:02 修改

阅读量678

点赞数

文章标签： unity 游戏引擎

于 2023-07-21 11:06:22 首次发布

本文链接：https://blog.csdn.net/weixin_44065829/article/details/131847591

版权

一. 一个compute文件对应一个函数

二. C#里使用该函数，三大步骤：

1. 开辟ComputeBuffer

2. 把数据传入到ComputBuffer里

3. 执行函数

4. 接收函数执行结果

5. 释放ComputeBuffer

//Compute Shader代码, 后缀为.compute文件
#pragma kernel AddVectors

RWStructuredBuffer<Vector4> inputA;
RWStructuredBuffer<Vector4> inputB;
RWStructuredBuffer<Vector4> output;

[numthreads(64, 1, 1)]
void AddVectors(uint3 id : SV_DispatchThreadID)
{
    output[id.x] = inputA[id.x] + inputB[id.x];
}

using UnityEngine;
using UnityEditor;

public class ComputeShaderExample : EditorWindow
{
    public ComputeShader computeShader;
    public int vectorCount = 1024;

    private ComputeBuffer inputBufferA;
    private ComputeBuffer inputBufferB;
    private ComputeBuffer outputBuffer;

    private void Calculate()
    {
        //