Class ArrivalToken

include/pipeline.h:12–40

enum class BarrierStatus : uint32_t { WaitAgain = 0u, WaitDone = 1u };

class ArrivalToken {
 public:
  HOST_DEVICE ArrivalToken(BarrierStatus barrier_status)
      : barrier_status_(barrier_status) {}

  HOST_DEVICE ArrivalToken() = delete;

  HOST_DEVICE BarrierStatus get() const {
    return barrier_status_;
    ;
  }

  HOST_DEVICE bool operator==(ArrivalToken const& other) const {
    return barrier_status_ == other.get();
  }

 private:
  BarrierStatus barrier_status_;

  HOST_DEVICE friend bool operator==(const ArrivalToken& left,
                                             const BarrierStatus& right) {
    return left.get() == right;
  }

  HOST_DEVICE friend bool operator==(const BarrierStatus& left,
                                             const ArrivalToken& right) {
    return left == right.get();
  }
};

class ProducerToken : public ArrivalToken {
  using ArrivalToken::ArrivalToken;