7b9e0fb428
This gets rather complicated because MSL does not support OpArrayLength natively. We need to pass down a buffer which contains buffer sizes, and we compute the array length on-demand. Support both discrete descriptors as well as argument buffers.
54 lines
1.3 KiB
Plaintext
54 lines
1.3 KiB
Plaintext
#include <metal_stdlib>
|
|
#include <simd/simd.h>
|
|
|
|
using namespace metal;
|
|
|
|
struct SSBO
|
|
{
|
|
uint size;
|
|
float4 v[1];
|
|
};
|
|
|
|
struct SSBO1
|
|
{
|
|
float bz[1];
|
|
};
|
|
|
|
struct SSBO2
|
|
{
|
|
uint size2;
|
|
float w[1];
|
|
};
|
|
|
|
struct SSBO3
|
|
{
|
|
float bz[1];
|
|
};
|
|
|
|
struct spvDescriptorSetBuffer0
|
|
{
|
|
constant uint* spvBufferSizeConstants [[id(0)]];
|
|
device SSBO* m_16 [[id(1)]];
|
|
};
|
|
|
|
struct spvDescriptorSetBuffer1
|
|
{
|
|
constant uint* spvBufferSizeConstants [[id(1)]];
|
|
device SSBO1* ssbos [[id(2)]][2];
|
|
};
|
|
|
|
kernel void main0(constant spvDescriptorSetBuffer0& spvDescriptorSet0 [[buffer(0)]], constant spvDescriptorSetBuffer1& spvDescriptorSet1 [[buffer(1)]], constant uint* spvBufferSizeConstants [[buffer(25)]], device SSBO3* ssbos2_0 [[buffer(2)]], device SSBO3* ssbos2_1 [[buffer(3)]])
|
|
{
|
|
device SSBO3* ssbos2[] =
|
|
{
|
|
ssbos2_0,
|
|
ssbos2_1,
|
|
};
|
|
|
|
constant uint& spvDescriptorSet0_m_16BufferSize = spvDescriptorSet0.spvBufferSizeConstants[1];
|
|
constant uint* spvDescriptorSet1_ssbosBufferSize = &spvDescriptorSet1.spvBufferSizeConstants[2];
|
|
constant uint* ssbos2BufferSize = &spvBufferSizeConstants[2];
|
|
(*spvDescriptorSet0.m_16).size = ((uint(int((spvDescriptorSet0_m_16BufferSize - 16) / 16)) + uint(int((spvDescriptorSet1_ssbosBufferSize[1] - 0) / 4))) + uint(int((_38BufferSize - 16) / 16))) + uint(int((ssbos2BufferSize[0] - 0) / 4));
|
|
}
|
|
|