TheRock/math-libs/libhipcxx/.upstream-tests/test/support/cmpxchg_loop.h

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include <cuda/std/atomic>

template <class A, class T>
__host__ __device__ bool cmpxchg_weak_loop(A& atomic, T& expected, T desired)
{
  for (int i = 0; i < 10; i++)
  {
    if (atomic.compare_exchange_weak(expected, desired) == true)
    {
      return true;
    }
  }

  return false;
}

template <class A, class T>
__host__ __device__ bool
cmpxchg_weak_loop(A& atomic, T& expected, T desired, cuda::std::memory_order success, cuda::std::memory_order failure)
{
  for (int i = 0; i < 10; i++)
  {
    if (atomic.compare_exchange_weak(expected, desired, success, failure) == true)
    {
      return true;
    }
  }

  return false;
}

template <class A, class T>
__host__ __device__ bool c_cmpxchg_weak_loop(A* atomic, T* expected, T desired)
{
  for (int i = 0; i < 10; i++)
  {
    if (cuda::std::atomic_compare_exchange_weak(atomic, expected, desired) == true)
    {
      return true;
    }
  }

  return false;
}

template <class A, class T>
__host__ __device__ bool
c_cmpxchg_weak_loop(A* atomic, T* expected, T desired, cuda::std::memory_order success, cuda::std::memory_order failure)
{
  for (int i = 0; i < 10; i++)
  {
    if (cuda::std::atomic_compare_exchange_weak_explicit(atomic, expected, desired, success, failure) == true)
    {
      return true;
    }
  }

  return false;
}