mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-05-14 19:31:25 +00:00
a5d73ce965
This is a major reworking of this transform. The old transform code was getting unwieldy, with part of the complication coming from the handling of multiple return statements. By generating a wrapper function instead, we can avoid a lot of this complexity. The original entry point function is stripped of all shader IO attributes (as well as `stage` and `workgroup_size`), but the body is left unmodified. A new entry point wrapper function is introduced which calls the original function, packing/unpacking the shader inputs as necessary, and propagates the result to the corresponding shader outputs. The new code has been refactored to use a state object with the different parts of the transform split into separate functions, which makes it much more manageable. Fixed: tint:1076 Bug: tint:920 Change-Id: I3490a0ea7a3509a4e198ce730e476516649d8d96 Reviewed-on: https://dawn-review.googlesource.com/c/tint/+/60521 Auto-Submit: James Price <jrprice@google.com> Kokoro: Kokoro <noreply+kokoro@google.com> Commit-Queue: James Price <jrprice@google.com> Reviewed-by: Ben Clayton <bclayton@google.com>
33 lines
897 B
Plaintext
33 lines
897 B
Plaintext
#include <metal_stdlib>
|
|
|
|
using namespace metal;
|
|
struct UBO {
|
|
/* 0x0000 */ int dynamic_idx;
|
|
};
|
|
struct tint_array_wrapper {
|
|
int arr[64];
|
|
};
|
|
struct S {
|
|
tint_array_wrapper data;
|
|
};
|
|
struct Result {
|
|
/* 0x0000 */ int out;
|
|
};
|
|
|
|
void f_inner(constant UBO& ubo, device Result& result, uint local_invocation_index, threadgroup S* const tint_symbol) {
|
|
for(uint idx = local_invocation_index; (idx < 64u); idx = (idx + 1u)) {
|
|
uint const i = idx;
|
|
(*(tint_symbol)).data.arr[i] = int();
|
|
}
|
|
threadgroup_barrier(mem_flags::mem_threadgroup);
|
|
(*(tint_symbol)).data.arr[ubo.dynamic_idx] = 1;
|
|
result.out = (*(tint_symbol)).data.arr[3];
|
|
}
|
|
|
|
kernel void f(uint local_invocation_index [[thread_index_in_threadgroup]], constant UBO& ubo [[buffer(0)]], device Result& result [[buffer(1)]]) {
|
|
threadgroup S tint_symbol_1;
|
|
f_inner(ubo, result, local_invocation_index, &(tint_symbol_1));
|
|
return;
|
|
}
|
|
|