Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 1 | struct strided_arr { |
| 2 | float el; |
| 3 | }; |
| 4 | struct strided_arr_1 { |
| 5 | strided_arr el[3][2]; |
| 6 | }; |
| 7 | |
Peng Huang | c00ff7f | 2023-03-31 17:55:19 +0000 | [diff] [blame] | 8 | RWByteAddressBuffer s : register(u0); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 9 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 10 | strided_arr s_load_4(uint offset) { |
| 11 | const strided_arr tint_symbol = {asfloat(s.Load((offset + 0u)))}; |
| 12 | return tint_symbol; |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 13 | } |
| 14 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 15 | typedef strided_arr s_load_3_ret[2]; |
| 16 | s_load_3_ret s_load_3(uint offset) { |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 17 | strided_arr arr[2] = (strided_arr[2])0; |
| 18 | { |
Antonio Maiorano | 06844a5 | 2022-09-29 16:53:58 +0000 | [diff] [blame] | 19 | for(uint i = 0u; (i < 2u); i = (i + 1u)) { |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 20 | arr[i] = s_load_4((offset + (i * 8u))); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 21 | } |
| 22 | } |
| 23 | return arr; |
| 24 | } |
| 25 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 26 | typedef strided_arr s_load_2_ret[3][2]; |
| 27 | s_load_2_ret s_load_2(uint offset) { |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 28 | strided_arr arr_1[3][2] = (strided_arr[3][2])0; |
| 29 | { |
Antonio Maiorano | 06844a5 | 2022-09-29 16:53:58 +0000 | [diff] [blame] | 30 | for(uint i_1 = 0u; (i_1 < 3u); i_1 = (i_1 + 1u)) { |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 31 | arr_1[i_1] = s_load_3((offset + (i_1 * 16u))); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 32 | } |
| 33 | } |
| 34 | return arr_1; |
| 35 | } |
| 36 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 37 | strided_arr_1 s_load_1(uint offset) { |
| 38 | const strided_arr_1 tint_symbol_1 = {s_load_2((offset + 0u))}; |
| 39 | return tint_symbol_1; |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 40 | } |
| 41 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 42 | typedef strided_arr_1 s_load_ret[4]; |
| 43 | s_load_ret s_load(uint offset) { |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 44 | strided_arr_1 arr_2[4] = (strided_arr_1[4])0; |
| 45 | { |
Antonio Maiorano | 06844a5 | 2022-09-29 16:53:58 +0000 | [diff] [blame] | 46 | for(uint i_2 = 0u; (i_2 < 4u); i_2 = (i_2 + 1u)) { |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 47 | arr_2[i_2] = s_load_1((offset + (i_2 * 128u))); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 48 | } |
| 49 | } |
| 50 | return arr_2; |
| 51 | } |
| 52 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 53 | void s_store_4(uint offset, strided_arr value) { |
| 54 | s.Store((offset + 0u), asuint(value.el)); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 55 | } |
| 56 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 57 | void s_store_3(uint offset, strided_arr value[2]) { |
Ben Clayton | 971318f | 2023-02-14 13:52:43 +0000 | [diff] [blame] | 58 | strided_arr array_3[2] = value; |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 59 | { |
Antonio Maiorano | 06844a5 | 2022-09-29 16:53:58 +0000 | [diff] [blame] | 60 | for(uint i_3 = 0u; (i_3 < 2u); i_3 = (i_3 + 1u)) { |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 61 | s_store_4((offset + (i_3 * 8u)), array_3[i_3]); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 62 | } |
| 63 | } |
| 64 | } |
| 65 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 66 | void s_store_2(uint offset, strided_arr value[3][2]) { |
Ben Clayton | 971318f | 2023-02-14 13:52:43 +0000 | [diff] [blame] | 67 | strided_arr array_2[3][2] = value; |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 68 | { |
Antonio Maiorano | 06844a5 | 2022-09-29 16:53:58 +0000 | [diff] [blame] | 69 | for(uint i_4 = 0u; (i_4 < 3u); i_4 = (i_4 + 1u)) { |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 70 | s_store_3((offset + (i_4 * 16u)), array_2[i_4]); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 71 | } |
| 72 | } |
| 73 | } |
| 74 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 75 | void s_store_1(uint offset, strided_arr_1 value) { |
| 76 | s_store_2((offset + 0u), value.el); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 77 | } |
| 78 | |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 79 | void s_store(uint offset, strided_arr_1 value[4]) { |
Ben Clayton | 971318f | 2023-02-14 13:52:43 +0000 | [diff] [blame] | 80 | strided_arr_1 array_1[4] = value; |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 81 | { |
Antonio Maiorano | 06844a5 | 2022-09-29 16:53:58 +0000 | [diff] [blame] | 82 | for(uint i_5 = 0u; (i_5 < 4u); i_5 = (i_5 + 1u)) { |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 83 | s_store_1((offset + (i_5 * 128u)), array_1[i_5]); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 84 | } |
| 85 | } |
| 86 | } |
| 87 | |
| 88 | void f_1() { |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 89 | const strided_arr_1 x_19[4] = s_load(0u); |
| 90 | const strided_arr x_24[3][2] = s_load_2(384u); |
| 91 | const strided_arr x_28[2] = s_load_3(416u); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 92 | const float x_32 = asfloat(s.Load(424u)); |
Ben Clayton | 1a1b527 | 2023-02-24 17:16:55 +0000 | [diff] [blame] | 93 | const strided_arr_1 tint_symbol_2[4] = (strided_arr_1[4])0; |
| 94 | s_store(0u, tint_symbol_2); |
Ben Clayton | 009d129 | 2022-02-04 15:39:34 +0000 | [diff] [blame] | 95 | s.Store(424u, asuint(5.0f)); |
| 96 | return; |
| 97 | } |
| 98 | |
| 99 | [numthreads(1, 1, 1)] |
| 100 | void f() { |
| 101 | f_1(); |
| 102 | return; |
| 103 | } |