//===----------------------------------------------------------------------===//
//
// 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/cstddef>

#include <test_macros.h>

// template <class IntegerType>
//   constexpr byte& operator>>=(byte& b, IntegerType shift) noexcept;
// This function shall not participate in overload resolution unless
//   is_integral_v<IntegerType> is true.

__host__ __device__ constexpr cuda::std::byte test(cuda::std::byte b)
{
  return b >>= 2;
}

int main(int, char**)
{
  cuda::std::byte b; // not constexpr, just used in noexcept check
  constexpr cuda::std::byte b16{static_cast<cuda::std::byte>(16)};
  constexpr cuda::std::byte b192{static_cast<cuda::std::byte>(192)};

  static_assert(noexcept(b >>= 2), "");

  assert(cuda::std::to_integer<int>(test(b16)) == 4);
  assert(cuda::std::to_integer<int>(test(b192)) == 48);

  static_assert(cuda::std::to_integer<int>(test(b16)) == 4, "");
  static_assert(cuda::std::to_integer<int>(test(b192)) == 48, "");

  return 0;
}
