Ben Clayton | 9dc48bc | 2023-01-24 14:55:17 +0000 | [diff] [blame] | 1 | alias ArrayType = array<vec4<i32>, 4>; |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 2 | |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 3 | struct S { |
James Price | 3b671cb | 2022-03-28 14:31:22 +0000 | [diff] [blame] | 4 | arr : ArrayType, |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 5 | }; |
| 6 | |
| 7 | var<private> src_private : ArrayType; |
| 8 | var<workgroup> src_workgroup : ArrayType; |
Ben Clayton | 01e4b6f | 2022-01-19 22:46:57 +0000 | [diff] [blame] | 9 | @group(0) @binding(0) var<uniform> src_uniform : S; |
| 10 | @group(0) @binding(1) var<storage, read_write> src_storage : S; |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 11 | |
| 12 | var<workgroup> dst : ArrayType; |
| 13 | var<workgroup> dst_nested : array<array<array<i32, 2>, 3>, 4>; |
| 14 | |
| 15 | fn ret_arr() -> ArrayType { |
| 16 | return ArrayType(); |
| 17 | } |
| 18 | |
| 19 | fn ret_struct_arr() -> S { |
| 20 | return S(); |
| 21 | } |
| 22 | |
| 23 | fn foo(src_param : ArrayType) { |
| 24 | var src_function : ArrayType; |
| 25 | |
Ben Clayton | 54a104e | 2023-02-22 20:04:40 +0000 | [diff] [blame] | 26 | // Assign from value constructor. |
James Price | f6e5cc0 | 2022-01-20 22:11:07 +0000 | [diff] [blame] | 27 | dst = ArrayType(vec4(1), vec4(2), vec4(3), vec4(3)); |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 28 | |
| 29 | // Assign from parameter. |
| 30 | dst = src_param; |
| 31 | |
| 32 | // Assign from function call. |
| 33 | dst = ret_arr(); |
| 34 | |
| 35 | // Assign from constant. |
| 36 | let src_let : ArrayType = ArrayType(); |
| 37 | dst = src_let; |
| 38 | |
dan sinclair | ff7cf21 | 2022-10-03 14:05:23 +0000 | [diff] [blame] | 39 | // Assign from var, various address spaces. |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 40 | dst = src_function; |
| 41 | dst = src_private; |
| 42 | dst = src_workgroup; |
| 43 | |
dan sinclair | ff7cf21 | 2022-10-03 14:05:23 +0000 | [diff] [blame] | 44 | // Assign from struct.arr, various address spaces. |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 45 | dst = ret_struct_arr().arr; |
| 46 | dst = src_uniform.arr; |
| 47 | dst = src_storage.arr; |
| 48 | |
| 49 | // Nested assignment. |
| 50 | var src_nested : array<array<array<i32, 2>, 3>, 4>; |
| 51 | dst_nested = src_nested; |
| 52 | } |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 53 | |
| 54 | @compute @workgroup_size(1) |
| 55 | fn main() { |
| 56 | let val = ArrayType(); |
| 57 | foo(val); |
| 58 | } |