/*
 * Copyright (c) 2022 NVIDIA Corporation
 *
 * Licensed under the Apache License Version 2.0 with LLVM Exceptions
 * (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *   https://llvm.org/LICENSE.txt
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

// clang-format Language: Cpp

#pragma once

#include "config.cuh"
#include <cuda/std/detail/__config>

#if _LIBCUDACXX_STD_VER > 17 && _LIBCUDACXX_CUDA_API_VERSION < 1009000
#  define STDEXEC_LIBCUDACXX_NEEDS_ATOMIC_WORKAROUND() 1
#else
#  define STDEXEC_LIBCUDACXX_NEEDS_ATOMIC_WORKAROUND() 0
#endif

#if STDEXEC_LIBCUDACXX_NEEDS_ATOMIC_WORKAROUND()

#  ifdef _CUDA_ATOMIC
#    error This file must be included before <cuda/atomic> from libcudacxx
#  endif

_LIBCUDACXX_BEGIN_NAMESPACE_STD

enum class memory_order : unsigned int;

namespace __detail
{

  template <class Tag>
  __device__ void __atomic_thread_fence_cuda(memory_order order, Tag);

  __device__ void __atomic_signal_fence_cuda(memory_order order);

#  if STDEXEC_CLANG()
  template <class Tag>
  __device__ void __atomic_thread_fence_cuda(int order, Tag);

  __device__ void __atomic_signal_fence_cuda(int order);
#  endif

  template <class Type, class Tag>
  __device__ Type __atomic_load_n_cuda(const volatile Type *ptr, memory_order order, Tag);

  template <class Type, class Tag>
  __device__ void __atomic_store_n_cuda(Type volatile *ptr, Type, memory_order order, Tag);

  template <class Type, class Tag>
  __device__ Type __atomic_exchange_n_cuda(Type volatile *ptr, Type val, memory_order order, Tag);

  template <class Type, class Tag>
  __device__ bool __atomic_compare_exchange_cuda(Type volatile *ptr,
                                                 Type          *expected,
                                                 Type const    *desired,
                                                 bool           weak,
                                                 memory_order   __success_memorder,
                                                 memory_order   __failure_memorder,
                                                 Tag);

}  // namespace __detail

_LIBCUDACXX_END_NAMESPACE_STD
#endif  // STDEXEC_LIBCUDACXX_NEEDS_ATOMIC_WORKAROUND()
