mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-17 08:46:41 +00:00
88 lines
3.3 KiB
C++
88 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>
|
|
|
|
_LIBCPP_BEGIN_NAMESPACE_STD
|
|
|
|
#if _LIBCPP_STD_VER >= 26
|
|
|
|
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;
|
|
}
|
|
};
|
|
|
|
#endif // _LIBCPP_STD_VER >= 26
|
|
|
|
_LIBCPP_END_NAMESPACE_STD
|
|
|
|
_LIBCPP_POP_MACROS
|
|
|
|
#endif // _LIBCPP___MDSPAN_ALIGNED_ACCESSOR_H
|