ad2fbea797
This has the benefit of consistent placement for these macros (outside the namespace, not inside it). This, in turn, makes it more obvious that the condition applies to the entire namespace scope. Compile times are also reduced a bit, since the compiler doesn't have to open a namespace just to close it again. This is enforced with a clang-tidy check. As a drive-by this also fixes a few macro comments.
87 lines
3.3 KiB
C++
87 lines
3.3 KiB
C++
// -*- C++ -*-
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
// Kokkos v. 4.0
|
|
// Copyright (2022) National Technology & Engineering
|
|
// Solutions of Sandia, LLC (NTESS).
|
|
//
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
|
// the U.S. Government retains certain rights in this software.
|
|
//
|
|
//===---------------------------------------------------------------------===//
|
|
|
|
#ifndef _LIBCPP___MDSPAN_ALIGNED_ACCESSOR_H
|
|
#define _LIBCPP___MDSPAN_ALIGNED_ACCESSOR_H
|
|
|
|
#include <__config>
|
|
#include <__cstddef/size_t.h>
|
|
#include <__mdspan/default_accessor.h>
|
|
#include <__memory/assume_aligned.h>
|
|
#include <__type_traits/is_abstract.h>
|
|
#include <__type_traits/is_array.h>
|
|
#include <__type_traits/is_convertible.h>
|
|
#include <__type_traits/remove_const.h>
|
|
|
|
#if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
|
|
# pragma GCC system_header
|
|
#endif
|
|
|
|
_LIBCPP_PUSH_MACROS
|
|
#include <__undef_macros>
|
|
|
|
#if _LIBCPP_STD_VER >= 26
|
|
|
|
_LIBCPP_BEGIN_NAMESPACE_STD
|
|
|
|
template <class _ElementType, size_t _ByteAlignment>
|
|
struct aligned_accessor {
|
|
static_assert(_ByteAlignment != 0 && (_ByteAlignment & (_ByteAlignment - 1)) == 0,
|
|
"aligned_accessor: byte alignment must be a power of two");
|
|
static_assert(_ByteAlignment >= alignof(_ElementType), "aligned_accessor: insufficient byte alignment");
|
|
static_assert(!is_array_v<_ElementType>, "aligned_accessor: template argument may not be an array type");
|
|
static_assert(!is_abstract_v<_ElementType>, "aligned_accessor: template argument may not be an abstract class");
|
|
|
|
using offset_policy = default_accessor<_ElementType>;
|
|
using element_type = _ElementType;
|
|
using reference = _ElementType&;
|
|
using data_handle_type = _ElementType*;
|
|
|
|
static constexpr size_t byte_alignment = _ByteAlignment;
|
|
|
|
_LIBCPP_HIDE_FROM_ABI constexpr aligned_accessor() noexcept = default;
|
|
|
|
template <class _OtherElementType, size_t _OtherByteAlignment>
|
|
requires(is_convertible_v<_OtherElementType (*)[], element_type (*)[]> && _OtherByteAlignment >= byte_alignment)
|
|
_LIBCPP_HIDE_FROM_ABI constexpr aligned_accessor(aligned_accessor<_OtherElementType, _OtherByteAlignment>) noexcept {}
|
|
|
|
template <class _OtherElementType>
|
|
requires(is_convertible_v<_OtherElementType (*)[], element_type (*)[]>)
|
|
_LIBCPP_HIDE_FROM_ABI explicit constexpr aligned_accessor(default_accessor<_OtherElementType>) noexcept {}
|
|
|
|
template <class _OtherElementType>
|
|
requires(is_convertible_v<element_type (*)[], _OtherElementType (*)[]>)
|
|
_LIBCPP_HIDE_FROM_ABI constexpr operator default_accessor<_OtherElementType>() const noexcept {
|
|
return {};
|
|
}
|
|
|
|
_LIBCPP_HIDE_FROM_ABI constexpr reference access(data_handle_type __p, size_t __i) const noexcept {
|
|
return std::assume_aligned<byte_alignment>(__p)[__i];
|
|
}
|
|
|
|
_LIBCPP_HIDE_FROM_ABI constexpr typename offset_policy::data_handle_type
|
|
offset(data_handle_type __p, size_t __i) const noexcept {
|
|
return std::assume_aligned<byte_alignment>(__p) + __i;
|
|
}
|
|
};
|
|
|
|
_LIBCPP_END_NAMESPACE_STD
|
|
|
|
#endif // _LIBCPP_STD_VER >= 26
|
|
|
|
_LIBCPP_POP_MACROS
|
|
|
|
#endif // _LIBCPP___MDSPAN_ALIGNED_ACCESSOR_H
|