Ben Clayton | 9dc42ad | 2024-02-05 18:01:33 +0000 | [diff] [blame] | 1 | groupshared int4 src_workgroup[4]; |
| 2 | |
| 3 | void tint_zero_workgroup_memory(uint local_idx) { |
| 4 | { |
| 5 | for(uint idx = local_idx; (idx < 4u); idx = (idx + 1u)) { |
| 6 | uint i = idx; |
| 7 | src_workgroup[i] = (0).xxxx; |
| 8 | } |
| 9 | } |
| 10 | GroupMemoryBarrierWithGroupSync(); |
| 11 | } |
| 12 | |
Ben Clayton | 4135ea5 | 2021-07-05 15:20:57 +0000 | [diff] [blame] | 13 | struct S { |
James Price | f6e5cc0 | 2022-01-20 22:11:07 +0000 | [diff] [blame] | 14 | int4 arr[4]; |
Ben Clayton | 5d2f34e | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 15 | }; |
| 16 | |
James Price | f6e5cc0 | 2022-01-20 22:11:07 +0000 | [diff] [blame] | 17 | static int4 src_private[4] = (int4[4])0; |
Peng Huang | c00ff7f | 2023-03-31 17:55:19 +0000 | [diff] [blame] | 18 | cbuffer cbuffer_src_uniform : register(b0) { |
Ben Clayton | 8a96c78 | 2021-07-15 20:29:09 +0000 | [diff] [blame] | 19 | uint4 src_uniform[4]; |
| 20 | }; |
Peng Huang | c00ff7f | 2023-03-31 17:55:19 +0000 | [diff] [blame] | 21 | RWByteAddressBuffer src_storage : register(u1); |
| 22 | RWByteAddressBuffer tint_symbol : register(u2); |
| 23 | RWByteAddressBuffer dst_nested : register(u3); |
Ben Clayton | 8a96c78 | 2021-07-15 20:29:09 +0000 | [diff] [blame] | 24 | |
James Price | f6e5cc0 | 2022-01-20 22:11:07 +0000 | [diff] [blame] | 25 | typedef int4 ret_arr_ret[4]; |
Ben Clayton | 8a96c78 | 2021-07-15 20:29:09 +0000 | [diff] [blame] | 26 | ret_arr_ret ret_arr() { |
Antonio Maiorano | 144ebed | 2024-01-30 16:30:15 +0000 | [diff] [blame] | 27 | int4 tint_symbol_4[4] = (int4[4])0; |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 28 | return tint_symbol_4; |
Ben Clayton | 5d2f34e | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 29 | } |
Ben Clayton | 9ef52ff | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 30 | |
Ben Clayton | 8a96c78 | 2021-07-15 20:29:09 +0000 | [diff] [blame] | 31 | S ret_struct_arr() { |
Antonio Maiorano | 144ebed | 2024-01-30 16:30:15 +0000 | [diff] [blame] | 32 | S tint_symbol_5 = (S)0; |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 33 | return tint_symbol_5; |
Ben Clayton | 165512c | 2021-06-18 21:15:25 +0000 | [diff] [blame] | 34 | } |
| 35 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 36 | void tint_symbol_store(uint offset, int4 value[4]) { |
Ben Clayton | 971318f | 2023-02-14 13:52:43 +0000 | [diff] [blame] | 37 | int4 array_1[4] = value; |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 38 | { |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 39 | for(uint i_1 = 0u; (i_1 < 4u); i_1 = (i_1 + 1u)) { |
| 40 | tint_symbol.Store4((offset + (i_1 * 16u)), asuint(array_1[i_1])); |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 41 | } |
| 42 | } |
Ben Clayton | 5d2f34e | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 43 | } |
| 44 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 45 | typedef int4 src_uniform_load_ret[4]; |
| 46 | src_uniform_load_ret src_uniform_load(uint offset) { |
James Price | f6e5cc0 | 2022-01-20 22:11:07 +0000 | [diff] [blame] | 47 | int4 arr_1[4] = (int4[4])0; |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 48 | { |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 49 | for(uint i_2 = 0u; (i_2 < 4u); i_2 = (i_2 + 1u)) { |
| 50 | const uint scalar_offset = ((offset + (i_2 * 16u))) / 4; |
| 51 | arr_1[i_2] = asint(src_uniform[scalar_offset / 4]); |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 52 | } |
| 53 | } |
| 54 | return arr_1; |
Ben Clayton | 8a96c78 | 2021-07-15 20:29:09 +0000 | [diff] [blame] | 55 | } |
| 56 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 57 | typedef int4 src_storage_load_ret[4]; |
| 58 | src_storage_load_ret src_storage_load(uint offset) { |
James Price | f6e5cc0 | 2022-01-20 22:11:07 +0000 | [diff] [blame] | 59 | int4 arr_2[4] = (int4[4])0; |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 60 | { |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 61 | for(uint i_3 = 0u; (i_3 < 4u); i_3 = (i_3 + 1u)) { |
| 62 | arr_2[i_3] = asint(src_storage.Load4((offset + (i_3 * 16u)))); |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 63 | } |
| 64 | } |
| 65 | return arr_2; |
Ben Clayton | 8a96c78 | 2021-07-15 20:29:09 +0000 | [diff] [blame] | 66 | } |
| 67 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 68 | void dst_nested_store_2(uint offset, int value[2]) { |
Ben Clayton | 971318f | 2023-02-14 13:52:43 +0000 | [diff] [blame] | 69 | int array_4[2] = value; |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 70 | { |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 71 | for(uint i_4 = 0u; (i_4 < 2u); i_4 = (i_4 + 1u)) { |
| 72 | dst_nested.Store((offset + (i_4 * 4u)), asuint(array_4[i_4])); |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 73 | } |
| 74 | } |
Ben Clayton | 5d2f34e | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 75 | } |
| 76 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 77 | void dst_nested_store_1(uint offset, int value[3][2]) { |
Ben Clayton | 971318f | 2023-02-14 13:52:43 +0000 | [diff] [blame] | 78 | int array_3[3][2] = value; |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 79 | { |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 80 | for(uint i_5 = 0u; (i_5 < 3u); i_5 = (i_5 + 1u)) { |
| 81 | dst_nested_store_2((offset + (i_5 * 8u)), array_3[i_5]); |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 82 | } |
| 83 | } |
Ben Clayton | 9ef52ff | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 84 | } |
| 85 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 86 | void dst_nested_store(uint offset, int value[4][3][2]) { |
Ben Clayton | 971318f | 2023-02-14 13:52:43 +0000 | [diff] [blame] | 87 | int array_2[4][3][2] = value; |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 88 | { |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 89 | for(uint i_6 = 0u; (i_6 < 4u); i_6 = (i_6 + 1u)) { |
| 90 | dst_nested_store_1((offset + (i_6 * 24u)), array_2[i_6]); |
Ben Clayton | 883fb63 | 2021-07-16 19:47:44 +0000 | [diff] [blame] | 91 | } |
| 92 | } |
Ben Clayton | 9ef52ff | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 93 | } |
| 94 | |
James Price | f6e5cc0 | 2022-01-20 22:11:07 +0000 | [diff] [blame] | 95 | void foo(int4 src_param[4]) { |
| 96 | int4 src_function[4] = (int4[4])0; |
Antonio Maiorano | 144ebed | 2024-01-30 16:30:15 +0000 | [diff] [blame] | 97 | int4 tint_symbol_6[4] = {(1).xxxx, (2).xxxx, (3).xxxx, (3).xxxx}; |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 98 | tint_symbol_store(0u, tint_symbol_6); |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 99 | tint_symbol_store(0u, src_param); |
Ben Clayton | da353b4 | 2023-03-08 21:48:45 +0000 | [diff] [blame] | 100 | tint_symbol_store(0u, ret_arr()); |
Antonio Maiorano | 144ebed | 2024-01-30 16:30:15 +0000 | [diff] [blame] | 101 | int4 src_let[4] = (int4[4])0; |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 102 | tint_symbol_store(0u, src_let); |
| 103 | tint_symbol_store(0u, src_function); |
| 104 | tint_symbol_store(0u, src_private); |
| 105 | tint_symbol_store(0u, src_workgroup); |
Antonio Maiorano | 144ebed | 2024-01-30 16:30:15 +0000 | [diff] [blame] | 106 | S tint_symbol_1 = ret_struct_arr(); |
Ben Clayton | da353b4 | 2023-03-08 21:48:45 +0000 | [diff] [blame] | 107 | tint_symbol_store(0u, tint_symbol_1.arr); |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 108 | tint_symbol_store(0u, src_uniform_load(0u)); |
| 109 | tint_symbol_store(0u, src_storage_load(0u)); |
Ben Clayton | 4135ea5 | 2021-07-05 15:20:57 +0000 | [diff] [blame] | 110 | int src_nested[4][3][2] = (int[4][3][2])0; |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 111 | dst_nested_store(0u, src_nested); |
Ben Clayton | 5d2f34e | 2021-06-16 09:19:36 +0000 | [diff] [blame] | 112 | } |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 113 | |
| 114 | struct tint_symbol_3 { |
| 115 | uint local_invocation_index : SV_GroupIndex; |
| 116 | }; |
| 117 | |
| 118 | void main_inner(uint local_invocation_index) { |
Ben Clayton | 9dc42ad | 2024-02-05 18:01:33 +0000 | [diff] [blame] | 119 | tint_zero_workgroup_memory(local_invocation_index); |
Antonio Maiorano | 144ebed | 2024-01-30 16:30:15 +0000 | [diff] [blame] | 120 | int4 ary[4] = (int4[4])0; |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 121 | foo(ary); |
| 122 | } |
| 123 | |
| 124 | [numthreads(1, 1, 1)] |
| 125 | void main(tint_symbol_3 tint_symbol_2) { |
| 126 | main_inner(tint_symbol_2.local_invocation_index); |
| 127 | return; |
| 128 | } |