forked from kokkos/kokkos
-
Notifications
You must be signed in to change notification settings - Fork 3
/
layout_padded_fwd.hpp
117 lines (100 loc) · 4.37 KB
/
layout_padded_fwd.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
//@HEADER
// ************************************************************************
//
// 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.
//
// Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
//
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//@HEADER
#pragma once
#include <cassert>
#include "../__p0009_bits/dynamic_extent.hpp"
namespace MDSPAN_IMPL_STANDARD_NAMESPACE {
namespace MDSPAN_IMPL_PROPOSED_NAMESPACE {
template <size_t padding_value = dynamic_extent>
struct layout_left_padded {
template <class _Extents>
class mapping;
};
template <size_t padding_value = dynamic_extent>
struct layout_right_padded {
template <class _Extents>
class mapping;
};
namespace detail {
// The layout_padded_constants structs are only useful if rank > 1, otherwise they may wrap
template <class _Layout, class _ExtentsType>
struct layout_padded_constants;
template <class _ExtentsType, size_t _PaddingStride>
struct layout_padded_constants<layout_left_padded<_PaddingStride>, _ExtentsType>
{
using rank_type = typename _ExtentsType::rank_type;
static constexpr rank_type padded_stride_idx = 1;
static constexpr rank_type extent_to_pad_idx = 0;
};
template <class _ExtentsType, size_t _PaddingStride>
struct layout_padded_constants<layout_right_padded<_PaddingStride>, _ExtentsType>
{
using rank_type = typename _ExtentsType::rank_type;
static constexpr rank_type padded_stride_idx = _ExtentsType::rank() - 2;
static constexpr rank_type extent_to_pad_idx = _ExtentsType::rank() - 1;
};
template <class _Layout>
struct is_layout_left_padded : std::false_type {};
template <size_t _PaddingStride>
struct is_layout_left_padded<layout_left_padded<_PaddingStride>> : std::true_type {};
template <class _Mapping, class _Enabled = void>
struct is_layout_left_padded_mapping : std::false_type {};
template <class _Mapping>
struct is_layout_left_padded_mapping<_Mapping,
std::enable_if_t<std::is_same<_Mapping, typename layout_left_padded<_Mapping::padding_value>::template mapping<typename _Mapping::extents_type>>::value>>
: std::true_type {};
template <class _Layout>
struct is_layout_right_padded : std::false_type {};
template <size_t _PaddingStride>
struct is_layout_right_padded<layout_right_padded<_PaddingStride>> : std::true_type {};
template <class _Mapping, class _Enabled = void>
struct is_layout_right_padded_mapping : std::false_type {};
template <class _Mapping>
struct is_layout_right_padded_mapping<_Mapping,
std::enable_if_t<std::is_same<_Mapping, typename layout_right_padded<_Mapping::padding_value>::template mapping<typename _Mapping::extents_type>>::value>>
: std::true_type {};
template <class _LayoutExtentsType, class _PaddedLayoutMappingType>
constexpr void check_padded_layout_converting_constructor_mandates()
{
if constexpr (_LayoutExtentsType::rank() > 1) {
using extents_type = typename _PaddedLayoutMappingType::extents_type;
constexpr auto padding_value = _PaddedLayoutMappingType::padding_value;
constexpr auto idx = layout_padded_constants<typename _PaddedLayoutMappingType::layout_type, _LayoutExtentsType >::extent_to_pad_idx;
if constexpr ((_LayoutExtentsType::static_extent(idx) != dynamic_extent) &&
(extents_type::static_extent(idx) != dynamic_extent) &&
(padding_value != dynamic_extent)) {
if constexpr (padding_value == 0) {
static_assert(_LayoutExtentsType::static_extent(idx) == 0);
} else {
static_assert(
_LayoutExtentsType::static_extent(idx) % padding_value == 0);
}
}
}
}
template <typename _ExtentsType, typename _OtherMapping>
constexpr void check_padded_layout_converting_constructor_preconditions([[maybe_unused]] const _OtherMapping &other_mapping) {
if constexpr (_ExtentsType::rank() > 1) {
constexpr auto padded_stride_idx =
layout_padded_constants<typename _OtherMapping::layout_type,
_ExtentsType>::padded_stride_idx;
constexpr auto extent_to_pad_idx = layout_padded_constants<typename _OtherMapping::layout_type, _ExtentsType>::extent_to_pad_idx;
assert(other_mapping.stride(padded_stride_idx) == other_mapping.extents().extent(extent_to_pad_idx));
}
}
}
}
}