rocm-systems/rocclr/runtime/thread/semaphore.hpp

//
// Copyright (c) 2008,2010 Advanced Micro Devices, Inc. All rights reserved.
//

#ifndef SEMAPHORE_HPP_
#define SEMAPHORE_HPP_

#include "top.hpp"
#include "utils/util.hpp"

#include <atomic>
#if defined(__linux__)
#include <semaphore.h>
#endif /*linux*/


namespace amd {

/*! \addtogroup Threads
 *  @{
 *
 *  \addtogroup Synchronization
 *  @{
 */

class Thread;

//! \brief Counting semaphore
class Semaphore : public HeapObject {
 private:
  std::atomic_int state_;  //!< This semaphore's value.

#ifdef _WIN32
  void* handle_;  //!< The semaphore object's handle.
  char padding_[64 - sizeof(void*) - sizeof(std::atomic_int)];
#else  // !_WIN32
  sem_t sem_;  //!< The semaphore object's identifier.
  char padding_[64 - sizeof(sem_t) - sizeof(std::atomic_int)];
#endif /*!_WIN32*/

 public:
  Semaphore();
  ~Semaphore();

  //! \brief Decrement this semaphore
  void wait();

  //! \brief Increment this semaphore
  void post();

  //! \brief Reset this semaphore.
  void reset() { state_.store(0, std::memory_order_release); }
};

/*! @}
 *  @}
 */

}  // namespace amd

#endif /*SEMAPHORE_HPP_*/