#include <metal_stdlib>
using namespace metal;

template<typename T, size_t N>
struct tint_array {
  const constant T& operator[](size_t i) const constant { return elements[i]; }
  device T& operator[](size_t i) device { return elements[i]; }
  const device T& operator[](size_t i) const device { return elements[i]; }
  thread T& operator[](size_t i) thread { return elements[i]; }
  const thread T& operator[](size_t i) const thread { return elements[i]; }
  threadgroup T& operator[](size_t i) threadgroup { return elements[i]; }
  const threadgroup T& operator[](size_t i) const threadgroup { return elements[i]; }
  T elements[N];
};

struct tint_immediate_data_struct {
  /* 0x0000 */ float2 user_immediate_data;
  /* 0x0008 */ uint tint_non_constant_zero;
  /* 0x000c */ tint_array<int8_t, 4> tint_pad;
};

struct tint_module_vars_struct {
  device float* out;
  const constant uint* idx;
  const constant tint_immediate_data_struct* tint_immediate_data;
};

[[max_total_threads_per_threadgroup(1)]]
kernel void v(device float* out [[buffer(0)]], const constant uint* idx [[buffer(1)]], const constant tint_immediate_data_struct* tint_immediate_data [[buffer(30)]]) {
  tint_module_vars_struct const tint_module_vars = tint_module_vars_struct{.out=out, .idx=idx, .tint_immediate_data=tint_immediate_data};
  (*tint_module_vars.out) = (*tint_module_vars.tint_immediate_data).user_immediate_data[min((*tint_module_vars.idx), 1u)];
}