2021-07-27 08:17:29 +00:00
|
|
|
#include <metal_stdlib>
|
|
|
|
|
|
|
|
using namespace metal;
|
2022-06-24 17:01:59 +00:00
|
|
|
|
|
|
|
template<typename T, size_t N>
|
|
|
|
struct tint_array {
|
|
|
|
const constant T& operator[](size_t i) const constant { return elements[i]; }
|
|
|
|
device T& operator[](size_t i) device { return elements[i]; }
|
|
|
|
const device T& operator[](size_t i) const device { return elements[i]; }
|
|
|
|
thread T& operator[](size_t i) thread { return elements[i]; }
|
|
|
|
const thread T& operator[](size_t i) const thread { return elements[i]; }
|
|
|
|
threadgroup T& operator[](size_t i) threadgroup { return elements[i]; }
|
|
|
|
const threadgroup T& operator[](size_t i) const threadgroup { return elements[i]; }
|
|
|
|
T elements[N];
|
2021-07-27 08:17:29 +00:00
|
|
|
};
|
2022-02-09 23:55:51 +00:00
|
|
|
|
2022-06-24 17:01:59 +00:00
|
|
|
struct strided_arr {
|
|
|
|
/* 0x0000 */ float2 el;
|
|
|
|
/* 0x0008 */ tint_array<int8_t, 8> tint_pad;
|
2021-07-27 08:17:29 +00:00
|
|
|
};
|
2022-02-09 23:55:51 +00:00
|
|
|
|
2021-07-27 08:17:29 +00:00
|
|
|
struct SSBO {
|
2022-06-24 17:01:59 +00:00
|
|
|
/* 0x0000 */ tint_array<strided_arr, 2> m;
|
2021-07-27 08:17:29 +00:00
|
|
|
};
|
|
|
|
|
2022-06-24 17:01:59 +00:00
|
|
|
float2x2 arr_to_mat2x2_stride_16(tint_array<strided_arr, 2> arr) {
|
|
|
|
return float2x2(arr[0u].el, arr[1u].el);
|
2021-07-27 08:17:29 +00:00
|
|
|
}
|
|
|
|
|
2022-06-24 17:01:59 +00:00
|
|
|
tint_array<strided_arr, 2> mat2x2_stride_16_to_arr(float2x2 m) {
|
2022-03-15 15:35:13 +00:00
|
|
|
strided_arr const tint_symbol_1 = {.el=m[0u]};
|
|
|
|
strided_arr const tint_symbol_2 = {.el=m[1u]};
|
2022-06-24 17:01:59 +00:00
|
|
|
tint_array<strided_arr, 2> const tint_symbol_3 = tint_array<strided_arr, 2>{tint_symbol_1, tint_symbol_2};
|
2022-03-15 15:35:13 +00:00
|
|
|
return tint_symbol_3;
|
2021-07-27 08:17:29 +00:00
|
|
|
}
|
|
|
|
|
2022-12-06 18:32:19 +00:00
|
|
|
void assign_and_preserve_padding_1(device strided_arr* const dest, strided_arr value) {
|
|
|
|
(*(dest)).el = value.el;
|
|
|
|
}
|
|
|
|
|
|
|
|
void assign_and_preserve_padding(device tint_array<strided_arr, 2>* const dest, tint_array<strided_arr, 2> value) {
|
|
|
|
for(uint i = 0u; (i < 2u); i = (i + 1u)) {
|
|
|
|
assign_and_preserve_padding_1(&((*(dest))[i]), value[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-03-15 15:35:13 +00:00
|
|
|
void f_1(device SSBO* const tint_symbol_4) {
|
|
|
|
float2x2 const x_15 = arr_to_mat2x2_stride_16((*(tint_symbol_4)).m);
|
2022-06-24 17:01:59 +00:00
|
|
|
tint_array<strided_arr, 2> const tint_symbol = mat2x2_stride_16_to_arr(x_15);
|
2022-12-06 18:32:19 +00:00
|
|
|
assign_and_preserve_padding(&((*(tint_symbol_4)).m), tint_symbol);
|
2021-07-27 08:17:29 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2022-03-15 15:35:13 +00:00
|
|
|
kernel void f(device SSBO* tint_symbol_5 [[buffer(0)]]) {
|
|
|
|
f_1(tint_symbol_5);
|
2021-07-27 08:17:29 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|