mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-12-10 05:57:51 +00:00
tint/writer/msl: Generate an array<T,N> helper
And remove the WrapArraysInStructs transform. Wrapping arrays in structures becomes troublesome for `const` arrays, as currently WGSL does not allow `const` structures. MSL 2.0+ has a builtin array<> helper, but we're targetting MSL 1.2, so we have to emit our own. Fortunately, it can be done with a few lines of templated code. This produces significantly cleaner output. Change-Id: Ifc92ef21e09befa252a07c856c4b5afdc51cc2e4 Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/94540 Kokoro: Kokoro <noreply+kokoro@google.com> Commit-Queue: Ben Clayton <bclayton@chromium.org> Reviewed-by: David Neto <dneto@google.com>
This commit is contained in:
committed by
Dawn LUCI CQ
parent
3c054304a8
commit
f47887d207
@@ -12,8 +12,16 @@ inline vec<T, N> operator*(packed_vec<T, M> lhs, matrix<T, N, M> rhs) {
|
||||
return vec<T, M>(lhs) * rhs;
|
||||
}
|
||||
|
||||
struct tint_array_wrapper {
|
||||
/* 0x0000 */ int4 arr[4];
|
||||
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];
|
||||
};
|
||||
|
||||
struct Inner {
|
||||
@@ -25,12 +33,12 @@ struct Inner {
|
||||
/* 0x002c */ float f;
|
||||
/* 0x0030 */ float2x3 g;
|
||||
/* 0x0050 */ float3x2 h;
|
||||
/* 0x0068 */ int8_t tint_pad[8];
|
||||
/* 0x0070 */ tint_array_wrapper i;
|
||||
/* 0x0068 */ tint_array<int8_t, 8> tint_pad;
|
||||
/* 0x0070 */ tint_array<int4, 4> i;
|
||||
};
|
||||
|
||||
struct S {
|
||||
/* 0x0000 */ Inner arr[1];
|
||||
/* 0x0000 */ tint_array<Inner, 1> arr;
|
||||
};
|
||||
|
||||
void tint_symbol_inner(uint idx, const device S* const tint_symbol_1) {
|
||||
@@ -42,7 +50,7 @@ void tint_symbol_inner(uint idx, const device S* const tint_symbol_1) {
|
||||
float const f = (*(tint_symbol_1)).arr[idx].f;
|
||||
float2x3 const g = (*(tint_symbol_1)).arr[idx].g;
|
||||
float3x2 const h = (*(tint_symbol_1)).arr[idx].h;
|
||||
tint_array_wrapper const i = (*(tint_symbol_1)).arr[idx].i;
|
||||
tint_array<int4, 4> const i = (*(tint_symbol_1)).arr[idx].i;
|
||||
}
|
||||
|
||||
kernel void tint_symbol(const device S* tint_symbol_2 [[buffer(0)]], uint idx [[thread_index_in_threadgroup]]) {
|
||||
|
||||
@@ -12,8 +12,16 @@ inline vec<T, N> operator*(packed_vec<T, M> lhs, matrix<T, N, M> rhs) {
|
||||
return vec<T, M>(lhs) * rhs;
|
||||
}
|
||||
|
||||
struct tint_array_wrapper {
|
||||
/* 0x0000 */ int4 arr[4];
|
||||
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];
|
||||
};
|
||||
|
||||
struct Inner {
|
||||
@@ -25,12 +33,12 @@ struct Inner {
|
||||
/* 0x002c */ float f;
|
||||
/* 0x0030 */ float2x3 g;
|
||||
/* 0x0050 */ float3x2 h;
|
||||
/* 0x0068 */ int8_t tint_pad[8];
|
||||
/* 0x0070 */ tint_array_wrapper i;
|
||||
/* 0x0068 */ tint_array<int8_t, 8> tint_pad;
|
||||
/* 0x0070 */ tint_array<int4, 4> i;
|
||||
};
|
||||
|
||||
struct S {
|
||||
/* 0x0000 */ Inner arr[1];
|
||||
/* 0x0000 */ tint_array<Inner, 1> arr;
|
||||
};
|
||||
|
||||
void tint_symbol_inner(uint idx, device S* const tint_symbol_2) {
|
||||
@@ -42,7 +50,7 @@ void tint_symbol_inner(uint idx, device S* const tint_symbol_2) {
|
||||
(*(tint_symbol_2)).arr[idx].f = 0.0f;
|
||||
(*(tint_symbol_2)).arr[idx].g = float2x3(float3(0.0f), float3(0.0f));
|
||||
(*(tint_symbol_2)).arr[idx].h = float3x2(float2(0.0f), float2(0.0f), float2(0.0f));
|
||||
tint_array_wrapper const tint_symbol_1 = {.arr={}};
|
||||
tint_array<int4, 4> const tint_symbol_1 = tint_array<int4, 4>{};
|
||||
(*(tint_symbol_2)).arr[idx].i = tint_symbol_1;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user