xref: /aosp_15_r20/external/skia/tests/sksl/compute/ModuleStructInExpression.metal (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1*c8dee2aaSAndroid Build Coastguard Worker#include <metal_stdlib>
2*c8dee2aaSAndroid Build Coastguard Worker#include <simd/simd.h>
3*c8dee2aaSAndroid Build Coastguard Worker#ifdef __clang__
4*c8dee2aaSAndroid Build Coastguard Worker#pragma clang diagnostic ignored "-Wall"
5*c8dee2aaSAndroid Build Coastguard Worker#endif
6*c8dee2aaSAndroid Build Coastguard Workerusing namespace metal;
7*c8dee2aaSAndroid Build Coastguard Workerstruct IndirectDispatchArgs {
8*c8dee2aaSAndroid Build Coastguard Worker    int x;
9*c8dee2aaSAndroid Build Coastguard Worker    int y;
10*c8dee2aaSAndroid Build Coastguard Worker    int z;
11*c8dee2aaSAndroid Build Coastguard Worker};
12*c8dee2aaSAndroid Build Coastguard Workerstruct Inputs {
13*c8dee2aaSAndroid Build Coastguard Worker};
14*c8dee2aaSAndroid Build Coastguard Workerstruct Threadgroups {
15*c8dee2aaSAndroid Build Coastguard Worker    int outX;
16*c8dee2aaSAndroid Build Coastguard Worker    int outY;
17*c8dee2aaSAndroid Build Coastguard Worker    int outZ;
18*c8dee2aaSAndroid Build Coastguard Worker};
19*c8dee2aaSAndroid Build Coastguard Workerint one_i() {
20*c8dee2aaSAndroid Build Coastguard Worker    return 1;
21*c8dee2aaSAndroid Build Coastguard Worker}
22*c8dee2aaSAndroid Build Coastguard Workerint two_i() {
23*c8dee2aaSAndroid Build Coastguard Worker    return 2;
24*c8dee2aaSAndroid Build Coastguard Worker}
25*c8dee2aaSAndroid Build Coastguard Workerint three_i() {
26*c8dee2aaSAndroid Build Coastguard Worker    return 3;
27*c8dee2aaSAndroid Build Coastguard Worker}
28*c8dee2aaSAndroid Build Coastguard Workerkernel void computeMain() {
29*c8dee2aaSAndroid Build Coastguard Worker    threadgroup Threadgroups _threadgroups{{}, {}, {}};
30*c8dee2aaSAndroid Build Coastguard Worker    (void)_threadgroups;
31*c8dee2aaSAndroid Build Coastguard Worker    Inputs _in = {  };
32*c8dee2aaSAndroid Build Coastguard Worker    _threadgroups.outX = IndirectDispatchArgs{one_i(), two_i(), three_i()}.x;
33*c8dee2aaSAndroid Build Coastguard Worker    _threadgroups.outY = IndirectDispatchArgs{one_i(), two_i(), three_i()}.y;
34*c8dee2aaSAndroid Build Coastguard Worker    _threadgroups.outZ = IndirectDispatchArgs{one_i(), two_i(), three_i()}.z;
35*c8dee2aaSAndroid Build Coastguard Worker    return;
36*c8dee2aaSAndroid Build Coastguard Worker}
37