Kokkos_LayoutNatural.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Sacado Package
//
// Copyright 2006 NTESS and the Sacado contributors.
// SPDX-License-Identifier: LGPL-2.1-or-later
// *****************************************************************************
// @HEADER

#ifndef KOKKOS_LAYOUT_NATURAL_HPP
#define KOKKOS_LAYOUT_NATURAL_HPP

// We are hooking into Kokkos Core internals here
// Need to define this macro since we include non-public headers
#ifndef KOKKOS_IMPL_PUBLIC_INCLUDE
#define KOKKOS_IMPL_PUBLIC_INCLUDE
#define KOKKOS_IMPL_PUBLIC_INCLUDE_NOTDEFINED_CORE
#endif
#include "Kokkos_Core_fwd.hpp"
#include "Kokkos_Layout.hpp"
#include "Kokkos_LayoutContiguous.hpp" // for inner_layout<>
#ifdef KOKKOS_IMPL_PUBLIC_INCLUDE_NOTDEFINED_CORE
#undef KOKKOS_IMPL_PUBLIC_INCLUDE
#undef KOKKOS_IMPL_PUBLIC_INCLUDE_NOTDEFINED_CORE
#endif

namespace Kokkos {

// Natural layout for scalar types -- equivalent to the wrapped
// layout type
template <typename Layout>
struct LayoutNatural : public Layout {

  typedef LayoutNatural array_layout ;

  LayoutNatural( LayoutNatural const & ) = default ;
  LayoutNatural( LayoutNatural && ) = default ;
  LayoutNatural & operator = ( LayoutNatural const & ) = default ;
  LayoutNatural & operator = ( LayoutNatural && ) = default ;

  KOKKOS_INLINE_FUNCTION
  constexpr LayoutNatural(
    size_t N0 = 0 , size_t N1 = 0 , size_t N2 = 0 , size_t N3 = 0
  , size_t N4 = 0 , size_t N5 = 0 , size_t N6 = 0 , size_t N7 = 0 )
    : Layout( N0 , N1 , N2 , N3 , N4 , N5 , N6 , N7 ) {}
};

// Is Layout == LayoutNatural<L> for some L
template <class Layout>
struct is_layout_natural {
  static const bool value = false;
};

template <class Layout>
struct is_layout_natural< LayoutNatural<Layout> > {
  static const bool value = true;
};

template <class Layout>
struct inner_layout< LayoutNatural<Layout> > {
  typedef Layout type;
};

} // namespace Kokkos

// Make LayoutNatural<Layout> equivalent to Layout
namespace std {

  template <class Layout>
  struct is_same< Kokkos::LayoutNatural<Layout>, Layout> {
    static const bool value = true;
  };

  template <class Layout>
  struct is_same< Layout, Kokkos::LayoutNatural<Layout> > {
    static const bool value = true;
  };

}

#if KOKKOS_VERSION >= 40499
#include "View/Kokkos_ViewMapping.hpp"
#else
#include "impl/Kokkos_ViewMapping.hpp"
#endif

namespace Kokkos {
namespace Impl {

// Implement ViewOffset for LayoutNatural
template < class Dimension , class Layout >
struct ViewOffset<Dimension, LayoutNatural<Layout>, void>
  : public ViewOffset<Dimension,Layout> {
public:

  // Would like to use inherited constructors, but gcc 4.7 doesn't support it
  //using ViewOffset<Dimension,Layout>::ViewOffset;

  typedef ViewOffset<Dimension,Layout> Base;

  ViewOffset() = default ;
  ViewOffset( const ViewOffset & ) = default ;
  ViewOffset & operator = ( const ViewOffset & ) = default ;

  // All constructors take one or two arguments

  template <typename Arg1>
  KOKKOS_INLINE_FUNCTION
  constexpr ViewOffset(const Arg1& arg1) : Base(arg1) {}

  template <typename Arg1, typename Arg2>
  KOKKOS_INLINE_FUNCTION
  constexpr ViewOffset(const Arg1& arg1, const Arg2& arg2) : Base(arg1,arg2) {}
};

} // namespace Impl
} // namespace Kokkos

#endif // #ifndef KOKKOS_LAYOUT_NATURAL_HPP