test/tint/statements/discard/atomic_cmpxchg.wgsl.expected.ir.msl

#include <metal_stdlib>
using namespace metal;

struct atomic_compare_exchange_result_i32 {
  int old_value;
  bool exchanged;
};

struct tint_module_vars_struct {
  device atomic_int* a;
  thread bool* continue_execution;
};

struct foo_outputs {
  int tint_symbol [[color(0)]];
};

atomic_compare_exchange_result_i32 v(device atomic_int* const atomic_ptr, int cmp, int val) {
  int old_value = cmp;
  bool const v_1 = atomic_compare_exchange_weak_explicit(atomic_ptr, (&old_value), val, memory_order_relaxed, memory_order_relaxed);
  return atomic_compare_exchange_result_i32{.old_value=old_value, .exchanged=v_1};
}

int foo_inner(tint_module_vars_struct tint_module_vars) {
  (*tint_module_vars.continue_execution) = false;
  int x = 0;
  atomic_compare_exchange_result_i32 v_2 = {};
  if ((*tint_module_vars.continue_execution)) {
    v_2 = v(tint_module_vars.a, 0, 1);
  }
  atomic_compare_exchange_result_i32 const result = v_2;
  if (result.exchanged) {
    x = result.old_value;
  }
  if (!((*tint_module_vars.continue_execution))) {
    discard_fragment();
  }
  return x;
}

fragment foo_outputs foo(device atomic_int* a [[buffer(0)]]) {
  thread bool continue_execution = true;
  tint_module_vars_struct const tint_module_vars = tint_module_vars_struct{.a=a, .continue_execution=(&continue_execution)};
  return foo_outputs{.tint_symbol=foo_inner(tint_module_vars)};
}