69 lines
2.0 KiB
Plaintext
69 lines
2.0 KiB
Plaintext
#pragma clang diagnostic ignored "-Wmissing-prototypes"
|
|
|
|
#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;
|
|
float4 w[1];
|
|
};
|
|
|
|
struct SSBO3
|
|
{
|
|
float bz[1];
|
|
};
|
|
|
|
constant uint3 gl_WorkGroupSize [[maybe_unused]] = uint3(1u);
|
|
|
|
struct spvDescriptorSetBuffer0
|
|
{
|
|
device SSBO* v_16 [[id(0)]];
|
|
constant uint* spvBufferSizeConstants [[id(1)]];
|
|
};
|
|
|
|
struct spvDescriptorSetBuffer1
|
|
{
|
|
device SSBO1* ssbos [[id(0)]][2];
|
|
constant uint* spvBufferSizeConstants [[id(2)]];
|
|
};
|
|
|
|
static inline __attribute__((always_inline))
|
|
uint get_size(device SSBO& v_16, constant uint& v_16BufferSize, device SSBO1* constant (&ssbos)[2], constant uint* ssbosBufferSize, device SSBO2& v_38, constant uint& v_38BufferSize, device SSBO3* (&ssbos2)[2], constant uint* ssbos2BufferSize)
|
|
{
|
|
uint len = uint(int((v_16BufferSize - 16) / 16));
|
|
len += uint(int((ssbosBufferSize[1] - 0) / 4));
|
|
len += uint(int((v_38BufferSize - 16) / 16));
|
|
len += uint(int((ssbos2BufferSize[0] - 0) / 4));
|
|
return len;
|
|
}
|
|
|
|
kernel void main0(constant spvDescriptorSetBuffer0& spvDescriptorSet0 [[buffer(0)]], constant spvDescriptorSetBuffer1& spvDescriptorSet1 [[buffer(1)]], constant uint* spvBufferSizeConstants [[buffer(25)]], device SSBO2& v_38 [[buffer(2)]], device SSBO3* ssbos2_0 [[buffer(3)]], device SSBO3* ssbos2_1 [[buffer(4)]])
|
|
{
|
|
device SSBO3* ssbos2[] =
|
|
{
|
|
ssbos2_0,
|
|
ssbos2_1,
|
|
};
|
|
|
|
constant uint& spvDescriptorSet0_v_16BufferSize = spvDescriptorSet0.spvBufferSizeConstants[0];
|
|
constant uint* spvDescriptorSet1_ssbosBufferSize = &spvDescriptorSet1.spvBufferSizeConstants[0];
|
|
constant uint& v_38BufferSize = spvBufferSizeConstants[2];
|
|
constant uint* ssbos2BufferSize = &spvBufferSizeConstants[3];
|
|
(*spvDescriptorSet0.v_16).size = get_size((*spvDescriptorSet0.v_16), spvDescriptorSet0_v_16BufferSize, spvDescriptorSet1.ssbos, spvDescriptorSet1_ssbosBufferSize, v_38, v_38BufferSize, ssbos2, ssbos2BufferSize);
|
|
}
|
|
|