James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 1 | #include <metal_stdlib> |
| 2 | |
| 3 | using namespace metal; |
Ben Clayton | f47887d | 2022-06-24 17:01:59 +0000 | [diff] [blame] | 4 | |
| 5 | template<typename T, size_t N> |
| 6 | struct tint_array { |
| 7 | const constant T& operator[](size_t i) const constant { return elements[i]; } |
| 8 | device T& operator[](size_t i) device { return elements[i]; } |
| 9 | const device T& operator[](size_t i) const device { return elements[i]; } |
| 10 | thread T& operator[](size_t i) thread { return elements[i]; } |
| 11 | const thread T& operator[](size_t i) const thread { return elements[i]; } |
| 12 | threadgroup T& operator[](size_t i) threadgroup { return elements[i]; } |
| 13 | const threadgroup T& operator[](size_t i) const threadgroup { return elements[i]; } |
| 14 | T elements[N]; |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 15 | }; |
Ben Clayton | 8ec32a6 | 2022-02-09 23:55:51 +0000 | [diff] [blame] | 16 | |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 17 | struct S { |
Ben Clayton | f47887d | 2022-06-24 17:01:59 +0000 | [diff] [blame] | 18 | tint_array<int, 4> arr; |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 19 | }; |
| 20 | |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 21 | int foo() { |
Ben Clayton | f47887d | 2022-06-24 17:01:59 +0000 | [diff] [blame] | 22 | tint_array<int, 4> const src = tint_array<int, 4>{}; |
| 23 | tint_array<int, 4> dst = {}; |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 24 | S dst_struct = {}; |
Ben Clayton | f47887d | 2022-06-24 17:01:59 +0000 | [diff] [blame] | 25 | tint_array<tint_array<int, 4>, 2> dst_array = {}; |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 26 | dst_struct.arr = src; |
Ben Clayton | f47887d | 2022-06-24 17:01:59 +0000 | [diff] [blame] | 27 | dst_array[1] = src; |
James Price | 567f2e4 | 2021-06-18 09:47:23 +0000 | [diff] [blame] | 28 | dst = src; |
| 29 | dst_struct.arr = src; |
Ben Clayton | f47887d | 2022-06-24 17:01:59 +0000 | [diff] [blame] | 30 | dst_array[0] = src; |
dan sinclair | a01fd24 | 2023-12-05 20:23:42 +0000 | [diff] [blame] | 31 | return as_type<int>((as_type<uint>(as_type<int>((as_type<uint>(dst[0]) + as_type<uint>(dst_struct.arr[0])))) + as_type<uint>(dst_array[0][0]))); |
| 32 | } |
| 33 | |
| 34 | kernel void tint_symbol(device int* tint_symbol_1 [[buffer(0)]]) { |
| 35 | *(tint_symbol_1) = foo(); |
| 36 | return; |
James Price | 94ac078 | 2021-06-02 17:23:03 +0000 | [diff] [blame] | 37 | } |
| 38 | |