128 lines
3.5 KiB
Plaintext
128 lines
3.5 KiB
Plaintext
// Generated by SharpEmu Gen5MslTranslator.
|
|
#include <metal_stdlib>
|
|
|
|
using namespace metal;
|
|
|
|
struct SharpEmuUniforms
|
|
{
|
|
uint dispatch_limit_x;
|
|
uint dispatch_limit_y;
|
|
uint dispatch_limit_z;
|
|
uint reserved;
|
|
uint buffer_bytes[1];
|
|
};
|
|
|
|
static inline uint sharpemu_load_word(device uint* b, uint bytes, uint addr)
|
|
{
|
|
if ((addr & 3u) == 0u)
|
|
{
|
|
return addr + 4u <= bytes ? b[addr >> 2] : 0u;
|
|
}
|
|
uint value = 0u;
|
|
device const uchar* p = (device const uchar*)b;
|
|
for (uint i = 0u; i < 4u; i++)
|
|
{
|
|
if (addr + i < bytes)
|
|
{
|
|
value |= (uint)p[addr + i] << (i * 8u);
|
|
}
|
|
}
|
|
return value;
|
|
}
|
|
|
|
static inline uint sharpemu_load_bytes(device uint* b, uint bytes, uint addr, uint count, bool signExtend)
|
|
{
|
|
uint value = 0u;
|
|
device const uchar* p = (device const uchar*)b;
|
|
for (uint i = 0u; i < count; i++)
|
|
{
|
|
if (addr + i < bytes)
|
|
{
|
|
value |= (uint)p[addr + i] << (i * 8u);
|
|
}
|
|
}
|
|
if (signExtend && count < 4u)
|
|
{
|
|
uint shift = 32u - (count * 8u);
|
|
value = (uint)(((int)(value << shift)) >> shift);
|
|
}
|
|
return value;
|
|
}
|
|
|
|
static inline void sharpemu_store_bytes(device uint* b, uint bytes, uint addr, uint value, uint count)
|
|
{
|
|
device uchar* p = (device uchar*)b;
|
|
for (uint i = 0u; i < count; i++)
|
|
{
|
|
if (addr + i < bytes)
|
|
{
|
|
p[addr + i] = (uchar)((value >> (i * 8u)) & 0xFFu);
|
|
}
|
|
}
|
|
}
|
|
|
|
static inline uint sharpemu_ballot(bool value)
|
|
{
|
|
return (uint)(uint64_t)simd_ballot(value);
|
|
}
|
|
|
|
static constant float sharpemu_off_i4_table[16] =
|
|
{
|
|
0.0f, 0.0625f, 0.1250f, 0.1875f, 0.2500f, 0.3125f, 0.3750f, 0.4375f,
|
|
-0.5000f, -0.4375f, -0.3750f, -0.3125f, -0.2500f, -0.1875f, -0.1250f, -0.0625f,
|
|
};
|
|
|
|
kernel void gen5_cs(
|
|
constant SharpEmuUniforms& sharpemu_uniforms [[buffer(0)]],
|
|
uint3 sharpemu_local_id [[thread_position_in_threadgroup]],
|
|
uint3 sharpemu_group_id [[threadgroup_position_in_grid]],
|
|
uint sharpemu_lane [[thread_index_in_simdgroup]])
|
|
{
|
|
uint s[128] = {};
|
|
uint v[256] = {};
|
|
bool exec = true;
|
|
bool vcc = false;
|
|
bool scc = false;
|
|
uint pc = 0u;
|
|
bool active = true;
|
|
uint steps = 0u;
|
|
uint bias[1] = {};
|
|
v[0] = sharpemu_local_id.x;
|
|
v[1] = sharpemu_local_id.y;
|
|
v[2] = sharpemu_local_id.z;
|
|
active = (sharpemu_group_id.x * 32u + sharpemu_local_id.x) < sharpemu_uniforms.dispatch_limit_x
|
|
&& (sharpemu_group_id.y * 1u + sharpemu_local_id.y) < sharpemu_uniforms.dispatch_limit_y
|
|
&& (sharpemu_group_id.z * 1u + sharpemu_local_id.z) < sharpemu_uniforms.dispatch_limit_z;
|
|
s[106] = 0u;
|
|
s[107] = 0u;
|
|
s[126] = sharpemu_ballot(true);
|
|
s[127] = 0u;
|
|
|
|
while (active)
|
|
{
|
|
switch (pc)
|
|
{
|
|
case 0u:
|
|
{
|
|
uint t0 = as_type<uint>((fma(as_type<float>(v[1]), as_type<float>(v[2]), as_type<float>(v[5]))));
|
|
if (exec) { v[5] = t0; }
|
|
uint t1 = as_type<uint>((fma(as_type<float>(v[1]), as_type<float>(0x42280000u), as_type<float>(v[2]))));
|
|
if (exec) { v[5] = t1; }
|
|
uint t2 = as_type<uint>((fma(as_type<float>(v[1]), as_type<float>(v[2]), as_type<float>(0x42280000u))));
|
|
if (exec) { v[5] = t2; }
|
|
uint t3 = as_type<uint>((fma(as_type<float>(v[1]), as_type<float>(v[2]), as_type<float>(v[5]))));
|
|
if (exec) { v[5] = t3; }
|
|
active = false;
|
|
}
|
|
break;
|
|
default:
|
|
active = false;
|
|
break;
|
|
}
|
|
if (++steps >= 100000u)
|
|
{
|
|
active = false;
|
|
}
|
|
}
|
|
}
|