Unity：如何直接在GPU上设置Mesh以避免CPU回传开销？

可以直接在GPU端完成Mesh构建，无需将顶点/索引数据回传至CPU，Unity提供了专门的API工作流，能直接利用Compute Shader中AppendStructuredBuffer生成的数据完成Mesh创建，彻底规避CPU回传的性能损耗。

核心实现思路 #

利用GraphicsBuffer作为GPU端数据容器，结合WritableMeshData系列API，直接将GPU中生成的顶点、索引数据拷贝到Mesh的显存区域，全程无需数据流转到CPU。

具体实现步骤 #

• Compute Shader端数据生成：声明AppendStructuredBuffer<VertexData>和AppendStructuredBuffer<uint>，在Compute Shader线程中生成并写入顶点、三角形索引数据。
• C#端缓冲区绑定：创建与Compute Shader中结构匹配的GraphicsBuffer，绑定到Compute Shader并执行dispatch，完成GPU端数据填充。
• Mesh数据容器分配：调用Mesh.AllocateWritableMeshData分配可写入的Mesh数据容器，设置顶点数量、拓扑结构等描述信息。
• GPU端数据拷贝：通过GraphicsBuffer.CopyCount获取实际生成的顶点/索引数量，再调用SetVertexBufferData和SetIndexBufferData直接从GraphicsBuffer拷贝数据到Mesh容器（此操作全程在GPU内完成）。
• Mesh数据应用：调用Mesh.ApplyAndDisposeWritableMeshData将处理后的Mesh数据应用到目标Mesh对象。

关键代码示例 #

C#端核心逻辑

using UnityEngine;

public struct VertexData
{
    [SerializeField] public Vector3 position;
    [SerializeField] public Vector3 normal;
}

public class GPUMeshBuilder : MonoBehaviour
{
    public ComputeShader meshComputeShader;
    public Mesh targetMesh;

    private GraphicsBuffer vertexBuffer;
    private GraphicsBuffer indexBuffer;

    void Start()
    {
        // 初始化GPU缓冲区
        int maxVertexCount = 10000;
        int maxIndexCount = 30000;
        vertexBuffer = new GraphicsBuffer(GraphicsBuffer.Target.Append, maxVertexCount, System.Runtime.InteropServices.Marshal.SizeOf(typeof(VertexData)));
        indexBuffer = new GraphicsBuffer(GraphicsBuffer.Target.Append, maxIndexCount, sizeof(uint));

        // 执行Compute Shader生成数据
        int kernel = meshComputeShader.FindKernel("CSMain");
        meshComputeShader.SetBuffer(kernel, "vertexBuffer", vertexBuffer);
        meshComputeShader.SetBuffer(kernel, "indexBuffer", indexBuffer);
        meshComputeShader.Dispatch(kernel, maxVertexCount / 64, 1, 1);

        // 获取实际生成的数据量
        int vertexCount = vertexBuffer.CopyCount;
        int indexCount = indexBuffer.CopyCount;

        // 分配并配置Mesh数据容器
        var meshData = Mesh.AllocateWritableMeshData(1);
        var meshDesc = new MeshDescriptor(vertexCount, indexCount, MeshTopology.Triangles)
        {
            vertexAttributes = new[] { VertexAttribute.Position, VertexAttribute.Normal }
        };
        meshData[0].SetDescriptor(meshDesc);

        // 直接从GPU缓冲区拷贝数据
        meshData[0].SetVertexBufferData(vertexBuffer, 0, 0, vertexCount, 0, GraphicsBuffer.Target.Append);
        meshData[0].SetIndexBufferData(indexBuffer, 0, 0, indexCount);

        // 应用数据到目标Mesh
        Mesh.ApplyAndDisposeWritableMeshData(meshData, targetMesh);

        // 释放缓冲区（按需管理生命周期）
        vertexBuffer.Release();
        indexBuffer.Release();
    }
}

Compute Shader端代码

#pragma kernel CSMain

struct VertexData
{
    float3 position;
    float3 normal;
};

AppendStructuredBuffer<VertexData> vertexBuffer;
AppendStructuredBuffer<uint> indexBuffer;

[numthreads(64,1,1)]
void CSMain(uint3 id : SV_DispatchThreadID)
{
    // 生成示例顶点数据
    VertexData v;
    v.position = float3(id.x * 0.1f, id.y * 0.1f, 0);
    v.normal = float3(0, 0, 1);
    vertexBuffer.Append(v);

    // 生成三角形索引（仅为示例，实际需根据拓扑逻辑生成）
    if (id.x < 9997)
    {
        indexBuffer.Append(id.x);
        indexBuffer.Append(id.x + 1);
        indexBuffer.Append(id.x + 2);
    }
}

注意事项 #

• 内存布局匹配：C#端的VertexData结构体必须与Compute Shader中的结构体内存布局完全一致，可通过[StructLayout(LayoutKind.Sequential)]标记确保对齐。
• GPU同步：调用CopyCount前需确保Compute Shader的dispatch执行完成，可通过Graphics.WaitForCompletion()强制同步（但会阻塞主线程，建议在异步流程中处理）。
• 版本兼容性：上述API仅在Unity 2019.3及以上版本支持，需确认项目使用的Unity版本符合要求。
• 缓冲区生命周期：GraphicsBuffer需手动调用Release()释放显存资源，避免内存泄漏。

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