// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project

#include <gtest/gtest.h>

#include <cstddef>
#include <sstream>
#include <iostream>

#include <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
import kokkos.core_impl;
#else
#include <Kokkos_Core.hpp>
#endif

namespace Test {

template <class Space>
void test_view_mapping() {
  using ExecSpace = typename Space::execution_space;

  using dim_0        = Kokkos::Impl::ViewDimension<>;
  using dim_s2       = Kokkos::Impl::ViewDimension<2>;
  using dim_s2_s3    = Kokkos::Impl::ViewDimension<2, 3>;
  using dim_s2_s3_s4 = Kokkos::Impl::ViewDimension<2, 3, 4>;

  using dim_s0       = Kokkos::Impl::ViewDimension<0>;
  using dim_s0_s3    = Kokkos::Impl::ViewDimension<0, 3>;
  using dim_s0_s3_s4 = Kokkos::Impl::ViewDimension<0, 3, 4>;

  using dim_s0_s0    = Kokkos::Impl::ViewDimension<0, 0>;
  using dim_s0_s0_s4 = Kokkos::Impl::ViewDimension<0, 0, 4>;

  using dim_s0_s0_s0          = Kokkos::Impl::ViewDimension<0, 0, 0>;
  using dim_s0_s0_s0_s0       = Kokkos::Impl::ViewDimension<0, 0, 0, 0>;
  using dim_s0_s0_s0_s0_s0    = Kokkos::Impl::ViewDimension<0, 0, 0, 0, 0>;
  using dim_s0_s0_s0_s0_s0_s0 = Kokkos::Impl::ViewDimension<0, 0, 0, 0, 0, 0>;
  using dim_s0_s0_s0_s0_s0_s0_s0 =
      Kokkos::Impl::ViewDimension<0, 0, 0, 0, 0, 0, 0>;
  using dim_s0_s0_s0_s0_s0_s0_s0_s0 =
      Kokkos::Impl::ViewDimension<0, 0, 0, 0, 0, 0, 0, 0>;

// Fully static dimensions should not be larger than an int.
#ifndef _WIN32  // For some reason on Windows the first test here fails with
                // size being 7 bytes on windows???
  ASSERT_LE(sizeof(dim_0), sizeof(int));
  ASSERT_LE(sizeof(dim_s2), sizeof(int));
  ASSERT_LE(sizeof(dim_s2_s3), sizeof(int));
  ASSERT_LE(sizeof(dim_s2_s3_s4), sizeof(int));

  // Rank 1 is size_t.
  ASSERT_EQ(sizeof(dim_s0), sizeof(size_t));
  ASSERT_EQ(sizeof(dim_s0_s3), sizeof(size_t));
  ASSERT_EQ(sizeof(dim_s0_s3_s4), sizeof(size_t));

  // Allow for padding.
  ASSERT_LE(sizeof(dim_s0_s0), 2 * sizeof(size_t));
  ASSERT_LE(sizeof(dim_s0_s0_s4), 2 * sizeof(size_t));

  ASSERT_LE(sizeof(dim_s0_s0_s0), 4 * sizeof(size_t));
  ASSERT_EQ(sizeof(dim_s0_s0_s0_s0), 4 * sizeof(unsigned));
  ASSERT_LE(sizeof(dim_s0_s0_s0_s0_s0), 6 * sizeof(unsigned));
  ASSERT_EQ(sizeof(dim_s0_s0_s0_s0_s0_s0), 6 * sizeof(unsigned));
  ASSERT_LE(sizeof(dim_s0_s0_s0_s0_s0_s0_s0), 8 * sizeof(unsigned));
  ASSERT_EQ(sizeof(dim_s0_s0_s0_s0_s0_s0_s0_s0), 8 * sizeof(unsigned));
#endif
  static_assert(int(dim_0::rank) == int(0));
  static_assert(int(dim_0::rank_dynamic) == int(0));
  static_assert(int(dim_0::ArgN0) == 1);
  static_assert(int(dim_0::ArgN1) == 1);
  static_assert(int(dim_0::ArgN2) == 1);

  static_assert(int(dim_s2::rank) == int(1));
  static_assert(int(dim_s2::rank_dynamic) == int(0));
  static_assert(int(dim_s2::ArgN0) == 2);
  static_assert(int(dim_s2::ArgN1) == 1);

  static_assert(int(dim_s2_s3::rank) == int(2));
  static_assert(int(dim_s2_s3::rank_dynamic) == int(0));
  static_assert(int(dim_s2_s3::ArgN0) == 2);
  static_assert(int(dim_s2_s3::ArgN1) == 3);
  static_assert(int(dim_s2_s3::ArgN2) == 1);

  static_assert(int(dim_s2_s3_s4::rank) == int(3));
  static_assert(int(dim_s2_s3_s4::rank_dynamic) == int(0));
  static_assert(int(dim_s2_s3_s4::ArgN0) == 2);
  static_assert(int(dim_s2_s3_s4::ArgN1) == 3);
  static_assert(int(dim_s2_s3_s4::ArgN2) == 4);
  static_assert(int(dim_s2_s3_s4::ArgN3) == 1);

  static_assert(int(dim_s0::rank) == int(1));
  static_assert(int(dim_s0::rank_dynamic) == int(1));

  static_assert(int(dim_s0_s3::rank) == int(2));
  static_assert(int(dim_s0_s3::rank_dynamic) == int(1));
  static_assert(int(dim_s0_s3::ArgN0) == 0);
  static_assert(int(dim_s0_s3::ArgN1) == 3);

  static_assert(int(dim_s0_s3_s4::rank) == int(3));
  static_assert(int(dim_s0_s3_s4::rank_dynamic) == int(1));
  static_assert(int(dim_s0_s3_s4::ArgN0) == 0);
  static_assert(int(dim_s0_s3_s4::ArgN1) == 3);
  static_assert(int(dim_s0_s3_s4::ArgN2) == 4);

  static_assert(int(dim_s0_s0_s4::rank) == int(3));
  static_assert(int(dim_s0_s0_s4::rank_dynamic) == int(2));
  static_assert(int(dim_s0_s0_s4::ArgN0) == 0);
  static_assert(int(dim_s0_s0_s4::ArgN1) == 0);
  static_assert(int(dim_s0_s0_s4::ArgN2) == 4);

  static_assert(int(dim_s0_s0_s0::rank) == int(3));
  static_assert(int(dim_s0_s0_s0::rank_dynamic) == int(3));

  static_assert(int(dim_s0_s0_s0_s0::rank) == int(4));
  static_assert(int(dim_s0_s0_s0_s0::rank_dynamic) == int(4));

  static_assert(int(dim_s0_s0_s0_s0_s0::rank) == int(5));
  static_assert(int(dim_s0_s0_s0_s0_s0::rank_dynamic) == int(5));

  static_assert(int(dim_s0_s0_s0_s0_s0_s0::rank) == int(6));
  static_assert(int(dim_s0_s0_s0_s0_s0_s0::rank_dynamic) == int(6));

  static_assert(int(dim_s0_s0_s0_s0_s0_s0_s0::rank) == int(7));
  static_assert(int(dim_s0_s0_s0_s0_s0_s0_s0::rank_dynamic) == int(7));

  static_assert(int(dim_s0_s0_s0_s0_s0_s0_s0_s0::rank) == int(8));
  static_assert(int(dim_s0_s0_s0_s0_s0_s0_s0_s0::rank_dynamic) == int(8));

  dim_s0 d1(2, 3, 4, 5, 6, 7, 8, 9);
  dim_s0_s0 d2(2, 3, 4, 5, 6, 7, 8, 9);
  dim_s0_s0_s0 d3(2, 3, 4, 5, 6, 7, 8, 9);
  dim_s0_s0_s0_s0 d4(2, 3, 4, 5, 6, 7, 8, 9);

  ASSERT_EQ(d1.N0, 2u);
  ASSERT_EQ(d2.N0, 2u);
  ASSERT_EQ(d3.N0, 2u);
  ASSERT_EQ(d4.N0, 2u);

  ASSERT_EQ(d1.N1, 1u);
  ASSERT_EQ(d2.N1, 3u);
  ASSERT_EQ(d3.N1, 3u);
  ASSERT_EQ(d4.N1, 3u);

  ASSERT_EQ(d1.N2, 1u);
  ASSERT_EQ(d2.N2, 1u);
  ASSERT_EQ(d3.N2, 4u);
  ASSERT_EQ(d4.N2, 4u);

  ASSERT_EQ(d1.N3, 1u);
  ASSERT_EQ(d2.N3, 1u);
  ASSERT_EQ(d3.N3, 1u);
  ASSERT_EQ(d4.N3, 5u);

  //----------------------------------------

  using stride_s0_s0_s0 =
      Kokkos::Impl::ViewOffset<dim_s0_s0_s0, Kokkos::LayoutStride>;

  //----------------------------------------
  // Static dimension.
  {
    using left_s2_s3_s4 =
        Kokkos::Impl::ViewOffset<dim_s2_s3_s4, Kokkos::LayoutLeft>;

    ASSERT_EQ(sizeof(left_s2_s3_s4), sizeof(dim_s2_s3_s4));

    left_s2_s3_s4 off3;

    stride_s0_s0_s0 stride3(off3);

    ASSERT_EQ(off3.stride_0(), 1u);
    ASSERT_EQ(off3.stride_1(), 2u);
    ASSERT_EQ(off3.stride_2(), 6u);
    ASSERT_EQ(off3.span(), 24u);

    ASSERT_EQ(off3.stride_0(), stride3.stride_0());
    ASSERT_EQ(off3.stride_1(), stride3.stride_1());
    ASSERT_EQ(off3.stride_2(), stride3.stride_2());
    ASSERT_EQ(off3.span(), stride3.span());

    unsigned offset = 0;

    for (int k = 0; k < 4; ++k)
      for (int j = 0; j < 3; ++j)
        for (int i = 0; i < 2; ++i, ++offset) {
          ASSERT_EQ(off3(i, j, k), offset);
          ASSERT_EQ(stride3(i, j, k), off3(i, j, k));
        }
  }

  //----------------------------------------
  // Small dimension is unpadded.
  {
    using left_s0_s0_s4 =
        Kokkos::Impl::ViewOffset<dim_s0_s0_s4, Kokkos::LayoutLeft>;

    left_s0_s0_s4 dyn_off3(std::integral_constant<unsigned, sizeof(int)>(),
                           Kokkos::LayoutLeft(2, 3, 0, 0, 0, 0, 0, 0));

    stride_s0_s0_s0 stride3(dyn_off3);

    ASSERT_EQ(dyn_off3.m_dim.rank, 3u);
    ASSERT_EQ(dyn_off3.m_dim.N0, 2u);
    ASSERT_EQ(dyn_off3.m_dim.N1, 3u);
    ASSERT_EQ(dyn_off3.m_dim.N2, 4u);
    ASSERT_EQ(dyn_off3.m_dim.N3, 1u);
    ASSERT_EQ(dyn_off3.size(), (size_t)2 * 3 * 4);

    const Kokkos::LayoutLeft layout = dyn_off3.layout();

    ASSERT_EQ(layout.dimension[0], 2u);
    ASSERT_EQ(layout.dimension[1], 3u);
    ASSERT_EQ(layout.dimension[2], 4u);
    ASSERT_EQ(layout.dimension[3], KOKKOS_INVALID_INDEX);
    ASSERT_EQ(layout.dimension[4], KOKKOS_INVALID_INDEX);
    ASSERT_EQ(layout.dimension[5], KOKKOS_INVALID_INDEX);
    ASSERT_EQ(layout.dimension[6], KOKKOS_INVALID_INDEX);
    ASSERT_EQ(layout.dimension[7], KOKKOS_INVALID_INDEX);

    ASSERT_EQ(stride3.m_dim.rank, 3u);
    ASSERT_EQ(stride3.m_dim.N0, 2u);
    ASSERT_EQ(stride3.m_dim.N1, 3u);
    ASSERT_EQ(stride3.m_dim.N2, 4u);
    ASSERT_EQ(stride3.m_dim.N3, 1u);
    ASSERT_EQ(stride3.size(), (size_t)2 * 3 * 4);

    size_t offset = 0;

    for (int k = 0; k < 4; ++k)
      for (int j = 0; j < 3; ++j)
        for (int i = 0; i < 2; ++i, ++offset) {
          ASSERT_EQ(offset, dyn_off3(i, j, k));
          ASSERT_EQ(stride3(i, j, k), dyn_off3(i, j, k));
        }

    ASSERT_EQ(dyn_off3.span(), offset);
    ASSERT_EQ(stride3.span(), dyn_off3.span());
  }

  //----------------------------------------
  // Large dimension is likely padded.
  {
    constexpr size_t N0 = 2000;
    constexpr size_t N1 = 300;

    using left_s0_s0_s4 =
        Kokkos::Impl::ViewOffset<dim_s0_s0_s4, Kokkos::LayoutLeft>;

    left_s0_s0_s4 dyn_off3(std::integral_constant<unsigned, sizeof(int)>(),
                           Kokkos::LayoutLeft(N0, N1, 0, 0, 0, 0, 0, 0));

    stride_s0_s0_s0 stride3(dyn_off3);

    ASSERT_EQ(dyn_off3.m_dim.rank, 3u);
    ASSERT_EQ(dyn_off3.m_dim.N0, N0);
    ASSERT_EQ(dyn_off3.m_dim.N1, N1);
    ASSERT_EQ(dyn_off3.m_dim.N2, 4u);
    ASSERT_EQ(dyn_off3.m_dim.N3, 1u);
    ASSERT_EQ(dyn_off3.size(), N0 * N1 * 4);

    ASSERT_EQ(stride3.m_dim.rank, 3u);
    ASSERT_EQ(stride3.m_dim.N0, N0);
    ASSERT_EQ(stride3.m_dim.N1, N1);
    ASSERT_EQ(stride3.m_dim.N2, 4u);
    ASSERT_EQ(stride3.m_dim.N3, 1u);
    ASSERT_EQ(stride3.size(), N0 * N1 * 4);
    ASSERT_EQ(stride3.span(), dyn_off3.span());

    size_t offset = 0;

    for (size_t k = 0; k < 4; ++k)
      for (size_t j = 0; j < N1; ++j)
        for (size_t i = 0; i < N0; ++i) {
          ASSERT_LE(offset, dyn_off3(i, j, k));
          ASSERT_EQ(stride3(i, j, k), dyn_off3(i, j, k));
          offset = dyn_off3(i, j, k) + 1;
        }

    ASSERT_LE(offset, dyn_off3.span());
  }

  //----------------------------------------
  // Static dimension.
  {
    using right_s2_s3_s4 =
        Kokkos::Impl::ViewOffset<dim_s2_s3_s4, Kokkos::LayoutRight>;

    ASSERT_EQ(sizeof(right_s2_s3_s4), sizeof(dim_s2_s3_s4));

    right_s2_s3_s4 off3;

    stride_s0_s0_s0 stride3(off3);

    ASSERT_EQ(off3.stride_0(), 12u);
    ASSERT_EQ(off3.stride_1(), 4u);
    ASSERT_EQ(off3.stride_2(), 1u);

    ASSERT_EQ(off3.dimension_0(), stride3.dimension_0());
    ASSERT_EQ(off3.dimension_1(), stride3.dimension_1());
    ASSERT_EQ(off3.dimension_2(), stride3.dimension_2());
    ASSERT_EQ(off3.stride_0(), stride3.stride_0());
    ASSERT_EQ(off3.stride_1(), stride3.stride_1());
    ASSERT_EQ(off3.stride_2(), stride3.stride_2());
    ASSERT_EQ(off3.span(), stride3.span());

    size_t offset = 0;

    for (int i = 0; i < 2; ++i)
      for (int j = 0; j < 3; ++j)
        for (int k = 0; k < 4; ++k, ++offset) {
          ASSERT_EQ(off3(i, j, k), offset);
          ASSERT_EQ(off3(i, j, k), stride3(i, j, k));
        }

    ASSERT_EQ(off3.span(), offset);
  }

  //----------------------------------------
  // Small dimension is unpadded.
  {
    using right_s0_s0_s4 =
        Kokkos::Impl::ViewOffset<dim_s0_s0_s4, Kokkos::LayoutRight>;

    right_s0_s0_s4 dyn_off3(std::integral_constant<unsigned, sizeof(int)>(),
                            Kokkos::LayoutRight(2, 3, 0, 0, 0, 0, 0, 0));

    stride_s0_s0_s0 stride3(dyn_off3);

    ASSERT_EQ(dyn_off3.m_dim.rank, 3u);
    ASSERT_EQ(dyn_off3.m_dim.N0, 2u);
    ASSERT_EQ(dyn_off3.m_dim.N1, 3u);
    ASSERT_EQ(dyn_off3.m_dim.N2, 4u);
    ASSERT_EQ(dyn_off3.m_dim.N3, 1u);
    ASSERT_EQ(dyn_off3.size(), (size_t)2 * 3 * 4);

    ASSERT_EQ(dyn_off3.dimension_0(), stride3.dimension_0());
    ASSERT_EQ(dyn_off3.dimension_1(), stride3.dimension_1());
    ASSERT_EQ(dyn_off3.dimension_2(), stride3.dimension_2());
    ASSERT_EQ(dyn_off3.stride_0(), stride3.stride_0());
    ASSERT_EQ(dyn_off3.stride_1(), stride3.stride_1());
    ASSERT_EQ(dyn_off3.stride_2(), stride3.stride_2());
    ASSERT_EQ(dyn_off3.span(), stride3.span());

    size_t offset = 0;

    for (int i = 0; i < 2; ++i)
      for (int j = 0; j < 3; ++j)
        for (int k = 0; k < 4; ++k, ++offset) {
          ASSERT_EQ(offset, dyn_off3(i, j, k));
          ASSERT_EQ(dyn_off3(i, j, k), stride3(i, j, k));
        }

    ASSERT_EQ(dyn_off3.span(), offset);
  }

  //----------------------------------------
  // Large dimension is likely padded.
  {
    constexpr size_t N0 = 2000;
    constexpr size_t N1 = 300;

    using right_s0_s0_s4 =
        Kokkos::Impl::ViewOffset<dim_s0_s0_s4, Kokkos::LayoutRight>;

    right_s0_s0_s4 dyn_off3(std::integral_constant<unsigned, sizeof(int)>(),
                            Kokkos::LayoutRight(N0, N1, 0, 0, 0, 0, 0, 0));

    stride_s0_s0_s0 stride3(dyn_off3);

    ASSERT_EQ(dyn_off3.m_dim.rank, 3u);
    ASSERT_EQ(dyn_off3.m_dim.N0, N0);
    ASSERT_EQ(dyn_off3.m_dim.N1, N1);
    ASSERT_EQ(dyn_off3.m_dim.N2, 4u);
    ASSERT_EQ(dyn_off3.m_dim.N3, 1u);
    ASSERT_EQ(dyn_off3.size(), N0 * N1 * 4);

    ASSERT_EQ(dyn_off3.dimension_0(), stride3.dimension_0());
    ASSERT_EQ(dyn_off3.dimension_1(), stride3.dimension_1());
    ASSERT_EQ(dyn_off3.dimension_2(), stride3.dimension_2());
    ASSERT_EQ(dyn_off3.stride_0(), stride3.stride_0());
    ASSERT_EQ(dyn_off3.stride_1(), stride3.stride_1());
    ASSERT_EQ(dyn_off3.stride_2(), stride3.stride_2());
    ASSERT_EQ(dyn_off3.span(), stride3.span());

    size_t offset = 0;

    for (size_t i = 0; i < N0; ++i)
      for (size_t j = 0; j < N1; ++j)
        for (size_t k = 0; k < 4; ++k) {
          ASSERT_LE(offset, dyn_off3(i, j, k));
          ASSERT_EQ(dyn_off3(i, j, k), stride3(i, j, k));
          offset = dyn_off3(i, j, k) + 1;
        }

    ASSERT_LE(offset, dyn_off3.span());
  }

  //----------------------------------------
  // Subview.
  {
    // Mapping rank 4 to rank 3
    using SubviewExtents = Kokkos::Impl::SubviewExtents<4, 3>;

    constexpr size_t N0 = 1000;
    constexpr size_t N1 = 2000;
    constexpr size_t N2 = 3000;
    constexpr size_t N3 = 4000;

    Kokkos::Impl::ViewDimension<N0, N1, N2, N3> dim;

    SubviewExtents tmp(dim, N0 / 2, Kokkos::ALL,
                       std::pair<size_t, size_t>(N2 / 4, 10 + N2 / 4),
                       Kokkos::pair<size_t, size_t>(N3 / 4, 20 + N3 / 4));

    ASSERT_EQ(tmp.domain_offset(0), N0 / 2);
    ASSERT_EQ(tmp.domain_offset(1), 0u);
    ASSERT_EQ(tmp.domain_offset(2), N2 / 4);
    ASSERT_EQ(tmp.domain_offset(3), N3 / 4);

    ASSERT_EQ(tmp.range_index(0), 1u);
    ASSERT_EQ(tmp.range_index(1), 2u);
    ASSERT_EQ(tmp.range_index(2), 3u);

    ASSERT_EQ(tmp.range_extent(0), N1);
    ASSERT_EQ(tmp.range_extent(1), 10u);
    ASSERT_EQ(tmp.range_extent(2), 20u);
  }

  {
    constexpr size_t N0 = 2000;
    constexpr size_t N1 = 300;

    constexpr size_t sub_N0 = 1000;
    constexpr size_t sub_N1 = 200;
    constexpr size_t sub_N2 = 4;

    using left_s0_s0_s4 =
        Kokkos::Impl::ViewOffset<dim_s0_s0_s4, Kokkos::LayoutLeft>;

    left_s0_s0_s4 dyn_off3(std::integral_constant<unsigned, sizeof(int)>(),
                           Kokkos::LayoutLeft(N0, N1, 0, 0, 0, 0, 0, 0));

    Kokkos::Impl::SubviewExtents<3, 3> sub(
        dyn_off3.m_dim, Kokkos::pair<int, int>(0, sub_N0),
        Kokkos::pair<int, int>(0, sub_N1), Kokkos::pair<int, int>(0, sub_N2));

    stride_s0_s0_s0 stride3(dyn_off3, sub);

    ASSERT_EQ(stride3.dimension_0(), sub_N0);
    ASSERT_EQ(stride3.dimension_1(), sub_N1);
    ASSERT_EQ(stride3.dimension_2(), sub_N2);
    ASSERT_EQ(stride3.size(), sub_N0 * sub_N1 * sub_N2);

    ASSERT_EQ(dyn_off3.stride_0(), stride3.stride_0());
    ASSERT_EQ(dyn_off3.stride_1(), stride3.stride_1());
    ASSERT_EQ(dyn_off3.stride_2(), stride3.stride_2());
    ASSERT_GE(dyn_off3.span(), stride3.span());

    for (size_t k = 0; k < sub_N2; ++k)
      for (size_t j = 0; j < sub_N1; ++j)
        for (size_t i = 0; i < sub_N0; ++i) {
          ASSERT_EQ(stride3(i, j, k), dyn_off3(i, j, k));
        }
  }

  {
    constexpr size_t N0 = 2000;
    constexpr size_t N1 = 300;

    constexpr size_t sub_N0 = 1000;
    constexpr size_t sub_N1 = 200;
    constexpr size_t sub_N2 = 4;

    using right_s0_s0_s4 =
        Kokkos::Impl::ViewOffset<dim_s0_s0_s4, Kokkos::LayoutRight>;

    right_s0_s0_s4 dyn_off3(std::integral_constant<unsigned, sizeof(int)>(),
                            Kokkos::LayoutRight(N0, N1, 0, 0, 0, 0, 0, 0));

    Kokkos::Impl::SubviewExtents<3, 3> sub(
        dyn_off3.m_dim, Kokkos::pair<int, int>(0, sub_N0),
        Kokkos::pair<int, int>(0, sub_N1), Kokkos::pair<int, int>(0, sub_N2));

    stride_s0_s0_s0 stride3(dyn_off3, sub);

    ASSERT_EQ(stride3.dimension_0(), sub_N0);
    ASSERT_EQ(stride3.dimension_1(), sub_N1);
    ASSERT_EQ(stride3.dimension_2(), sub_N2);
    ASSERT_EQ(stride3.size(), sub_N0 * sub_N1 * sub_N2);

    ASSERT_EQ(dyn_off3.stride_0(), stride3.stride_0());
    ASSERT_EQ(dyn_off3.stride_1(), stride3.stride_1());
    ASSERT_EQ(dyn_off3.stride_2(), stride3.stride_2());
    ASSERT_GE(dyn_off3.span(), stride3.span());

    for (size_t i = 0; i < sub_N0; ++i)
      for (size_t j = 0; j < sub_N1; ++j)
        for (size_t k = 0; k < sub_N2; ++k) {
          ASSERT_EQ(stride3(i, j, k), dyn_off3(i, j, k));
        }
  }

  //----------------------------------------
  // View data analysis.
  {
    using namespace Kokkos::Impl;

    static_assert(rank_dynamic<>::value == 0);
    static_assert(rank_dynamic<1>::value == 0);
    static_assert(rank_dynamic<0>::value == 1);
    static_assert(rank_dynamic<0, 1>::value == 1);
    static_assert(rank_dynamic<0, 0, 1>::value == 2);
  }

  {
    using namespace Kokkos::Impl;

    using a_int_r1       = ViewArrayAnalysis<int[]>;
    using a_int_r5       = ViewArrayAnalysis<int** [4][5][6]>;
    using a_const_int_r1 = ViewArrayAnalysis<const int[]>;
    using a_const_int_r5 = ViewArrayAnalysis<const int** [4][5][6]>;

    static_assert(a_int_r1::dimension::rank == 1);
    static_assert(a_int_r1::dimension::rank_dynamic == 1);
    static_assert(a_int_r5::dimension::ArgN0 == 0);
    static_assert(a_int_r5::dimension::ArgN1 == 0);
    static_assert(a_int_r5::dimension::ArgN2 == 4);
    static_assert(a_int_r5::dimension::ArgN3 == 5);
    static_assert(a_int_r5::dimension::ArgN4 == 6);
    static_assert(a_int_r5::dimension::ArgN5 == 1);

    static_assert(
        std::is_same_v<typename a_int_r1::dimension, ViewDimension<0> >);
    static_assert(std::is_same_v<typename a_int_r1::non_const_value_type, int>);

    static_assert(a_const_int_r1::dimension::rank == 1);
    static_assert(a_const_int_r1::dimension::rank_dynamic == 1);
    static_assert(
        std::is_same_v<typename a_const_int_r1::dimension, ViewDimension<0> >);
    static_assert(
        std::is_same_v<typename a_const_int_r1::non_const_value_type, int>);

    static_assert(a_const_int_r5::dimension::rank == 5);
    static_assert(a_const_int_r5::dimension::rank_dynamic == 2);

    static_assert(a_const_int_r5::dimension::ArgN0 == 0);
    static_assert(a_const_int_r5::dimension::ArgN1 == 0);
    static_assert(a_const_int_r5::dimension::ArgN2 == 4);
    static_assert(a_const_int_r5::dimension::ArgN3 == 5);
    static_assert(a_const_int_r5::dimension::ArgN4 == 6);
    static_assert(a_const_int_r5::dimension::ArgN5 == 1);

    static_assert(std::is_same_v<typename a_const_int_r5::dimension,
                                 ViewDimension<0, 0, 4, 5, 6> >);
    static_assert(
        std::is_same_v<typename a_const_int_r5::non_const_value_type, int>);

    static_assert(a_int_r5::dimension::rank == 5);
    static_assert(a_int_r5::dimension::rank_dynamic == 2);
    static_assert(std::is_same_v<typename a_int_r5::dimension,
                                 ViewDimension<0, 0, 4, 5, 6> >);
    static_assert(std::is_same_v<typename a_int_r5::non_const_value_type, int>);
  }

  {
    using namespace Kokkos::Impl;

    using t_i4 = int[4];

    // Dimensions of t_i4 are appended to the multdimensional array.
    using a_int_r5 = ViewArrayAnalysis<t_i4*** [3]>;

    static_assert(a_int_r5::dimension::rank == 5);
    static_assert(a_int_r5::dimension::rank_dynamic == 3);
    static_assert(a_int_r5::dimension::ArgN0 == 0);
    static_assert(a_int_r5::dimension::ArgN1 == 0);
    static_assert(a_int_r5::dimension::ArgN2 == 0);
    static_assert(a_int_r5::dimension::ArgN3 == 3);
    static_assert(a_int_r5::dimension::ArgN4 == 4);
    static_assert(std::is_same_v<typename a_int_r5::non_const_value_type, int>);
  }

  {
    using namespace Kokkos::Impl;

    using a_const_int_r1 = ViewDataAnalysis<const int[], void>;

    static_assert(std::is_void_v<typename a_const_int_r1::specialize>);
    static_assert(std::is_same_v<typename a_const_int_r1::dimension,
                                 Kokkos::Impl::ViewDimension<0> >);
    static_assert(
        std::is_same_v<typename a_const_int_r1::data_type, const int*>);
    static_assert(
        std::is_same_v<typename a_const_int_r1::const_data_type, const int*>);
    static_assert(
        std::is_same_v<typename a_const_int_r1::non_const_data_type, int*>);
    static_assert(
        std::is_same_v<typename a_const_int_r1::value_type, const int>);
    static_assert(
        std::is_same_v<typename a_const_int_r1::const_value_type, const int>);
    static_assert(
        std::is_same_v<typename a_const_int_r1::non_const_value_type, int>);
#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_PUSH()
    static_assert(
        std::is_same_v<typename a_const_int_r1::scalar_array_type, const int*>);
    static_assert(
        std::is_same_v<typename a_const_int_r1::const_scalar_array_type,
                       const int*>);
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_POP()
#endif

    using a_const_int_r3 = ViewDataAnalysis<const int** [4], void>;

    static_assert(std::is_void_v<typename a_const_int_r3::specialize>);
    static_assert(std::is_same_v<typename a_const_int_r3::dimension,
                                 Kokkos::Impl::ViewDimension<0, 0, 4> >);
    static_assert(
        std::is_same_v<typename a_const_int_r3::data_type, const int** [4]>);
    static_assert(std::is_same_v<typename a_const_int_r3::const_data_type,
                                 const int** [4]>);
    static_assert(std::is_same_v<typename a_const_int_r3::non_const_data_type,
                                 int** [4]>);
    static_assert(
        std::is_same_v<typename a_const_int_r3::value_type, const int>);
    static_assert(
        std::is_same_v<typename a_const_int_r3::const_value_type, const int>);
    static_assert(
        std::is_same_v<typename a_const_int_r3::non_const_value_type, int>);
#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_PUSH()
    static_assert(std::is_same_v<typename a_const_int_r3::scalar_array_type,
                                 const int** [4]>);
    static_assert(
        std::is_same_v<typename a_const_int_r3::const_scalar_array_type,
                       const int** [4]>);
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_POP()
#endif
  }

  //----------------------------------------

  {
    constexpr int N = 10;

    using T = Kokkos::View<int*, Space>;
    using C = Kokkos::View<const int*, Space>;

    static_assert(std::is_same_v<typename T::data_type, int*>);
    static_assert(std::is_same_v<typename T::const_data_type, const int*>);
    static_assert(std::is_same_v<typename T::non_const_data_type, int*>);

#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_PUSH()
    static_assert(std::is_same_v<typename T::scalar_array_type, int*>);
    static_assert(
        std::is_same_v<typename T::const_scalar_array_type, const int*>);
    static_assert(
        std::is_same_v<typename T::non_const_scalar_array_type, int*>);
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_POP()
#endif

    static_assert(std::is_same_v<typename T::value_type, int>);
    static_assert(std::is_same_v<typename T::const_value_type, const int>);
    static_assert(std::is_same_v<typename T::non_const_value_type, int>);

    static_assert(
        std::is_same_v<typename T::memory_space, typename Space::memory_space>);
    static_assert(std::is_same_v<typename T::reference_type, int&>);

    static_assert(T::rank == size_t(1));

    static_assert(std::is_same_v<typename C::data_type, const int*>);
    static_assert(std::is_same_v<typename C::const_data_type, const int*>);
    static_assert(std::is_same_v<typename C::non_const_data_type, int*>);

#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_PUSH()
    static_assert(std::is_same_v<typename C::scalar_array_type, const int*>);
    static_assert(
        std::is_same_v<typename C::const_scalar_array_type, const int*>);
    static_assert(
        std::is_same_v<typename C::non_const_scalar_array_type, int*>);
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_POP()
#endif
    static_assert(std::is_same_v<typename C::value_type, const int>);
    static_assert(std::is_same_v<typename C::const_value_type, const int>);
    static_assert(std::is_same_v<typename C::non_const_value_type, int>);

    static_assert(
        std::is_same_v<typename C::memory_space, typename Space::memory_space>);
    static_assert(std::is_same_v<typename C::reference_type, const int&>);

    static_assert(C::rank == size_t(1));

    if (Kokkos::SpaceAccessibility<Kokkos::HostSpace,
                                   typename Space::memory_space>::accessible &&
        Kokkos::SpaceAccessibility<typename Space::memory_space,
                                   Kokkos::HostSpace>::assignable) {
      int data[N];

      T vr1(data, N);  // View of non-const.
      C cr1(vr1);      // View of const from view of non-const.
      C cr2((const int*)data, N);

      // Generate static_assert error:
      // T tmp( cr1 );

      ASSERT_EQ(vr1.span(), size_t(N));
      ASSERT_EQ(cr1.span(), size_t(N));
      ASSERT_EQ(vr1.data(), &data[0]);
      ASSERT_EQ(cr1.data(), &data[0]);

      ASSERT_EQ(vr1.extent(0), size_t(N));

      for (int i = 0; i < N; ++i) data[i] = i + 1;
      for (int i = 0; i < N; ++i) ASSERT_EQ(vr1[i], i + 1);
      for (int i = 0; i < N; ++i) ASSERT_EQ(cr1[i], i + 1);

      {
        T tmp(vr1);

        for (int i = 0; i < N; ++i) ASSERT_EQ(tmp[i], i + 1);
        for (int i = 0; i < N; ++i) vr1(i) = i + 2;
        for (int i = 0; i < N; ++i) ASSERT_EQ(tmp[i], i + 2);
      }

      for (int i = 0; i < N; ++i) ASSERT_EQ(vr1[i], i + 2);
    }
  }

  {
    constexpr int N = 10;
    using T         = Kokkos::View<int*, Space>;
    using C         = Kokkos::View<const int*, Space>;

    T vr1("vr1", N);
    C cr1(vr1);

    static_assert(std::is_same_v<typename T::data_type, int*>);
    static_assert(std::is_same_v<typename T::const_data_type, const int*>);
    static_assert(std::is_same_v<typename T::non_const_data_type, int*>);

#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_PUSH()
    static_assert(std::is_same_v<typename T::scalar_array_type, int*>);
    static_assert(
        std::is_same_v<typename T::const_scalar_array_type, const int*>);
    static_assert(
        std::is_same_v<typename T::non_const_scalar_array_type, int*>);
    KOKKOS_IMPL_DISABLE_DEPRECATED_WARNINGS_POP()
#endif

    static_assert(std::is_same_v<typename T::value_type, int>);
    static_assert(std::is_same_v<typename T::const_value_type, const int>);
    static_assert(std::is_same_v<typename T::non_const_value_type, int>);

    static_assert(
        std::is_same_v<typename T::memory_space, typename Space::memory_space>);
    static_assert(std::is_same_v<typename T::reference_type, int&>);
    static_assert(T::rank == size_t(1));

    ASSERT_EQ(vr1.extent(0), size_t(N));

    if (Kokkos::SpaceAccessibility<Kokkos::HostSpace,
                                   typename Space::memory_space>::accessible) {
      for (int i = 0; i < N; ++i) vr1(i) = i + 1;
      for (int i = 0; i < N; ++i) ASSERT_EQ(vr1[i], i + 1);
      for (int i = 0; i < N; ++i) ASSERT_EQ(cr1[i], i + 1);

      {
        T tmp(vr1);
        for (int i = 0; i < N; ++i) ASSERT_EQ(tmp[i], i + 1);
        for (int i = 0; i < N; ++i) vr1(i) = i + 2;
        for (int i = 0; i < N; ++i) ASSERT_EQ(tmp[i], i + 2);
      }

      for (int i = 0; i < N; ++i) ASSERT_EQ(vr1[i], i + 2);
    }
  }

  // Testing proper handling of zero-length allocations.
  {
    constexpr int N = 0;
    using T         = Kokkos::View<int*, Space>;
    using C         = Kokkos::View<const int*, Space>;

    T vr1("vr1", N);
    C cr1(vr1);

    ASSERT_EQ(vr1.extent(0), 0u);
    ASSERT_EQ(cr1.extent(0), 0u);
  }

  // Testing using space instance for allocation.
  // The execution space of the memory space must be available for view data
  // initialization.
  if (std::is_same_v<ExecSpace,
                     typename ExecSpace::memory_space::execution_space>) {
    using namespace Kokkos;

    using memory_space = typename ExecSpace::memory_space;
    using V            = View<int*, memory_space>;

    constexpr int N = 10;

    memory_space mem_space;

    V v("v", N);
    V va(view_alloc(), N);
    V vb(view_alloc("vb"), N);
    V vc(view_alloc("vc", AllowPadding), N);
    V vd(view_alloc("vd", WithoutInitializing), N);
    V ve(view_alloc("ve", WithoutInitializing, AllowPadding), N);
    V vf(view_alloc("vf", mem_space, WithoutInitializing, AllowPadding), N);
    V vg(view_alloc(mem_space, "vg", WithoutInitializing, AllowPadding), N);
    V vh(view_alloc(WithoutInitializing, AllowPadding), N);
    V vi(view_alloc(WithoutInitializing), N);
    V vj(view_alloc(std::string("vj"), AllowPadding), N);
    V vk(view_alloc(mem_space, std::string("vk"), AllowPadding), N);
  }

  {
    using traits_t =
        Kokkos::ViewTraits<int***, Kokkos::LayoutStride, ExecSpace>;
    using dims_t   = Kokkos::Impl::ViewDimension<0, 0, 0>;
    using offset_t = Kokkos::Impl::ViewOffset<dims_t, Kokkos::LayoutStride>;

    Kokkos::LayoutStride stride;

    stride.dimension[0] = 3;
    stride.dimension[1] = 4;
    stride.dimension[2] = 5;
    stride.stride[0]    = 4;
    stride.stride[1]    = 1;
    stride.stride[2]    = 12;

    const offset_t offset(std::integral_constant<unsigned, 0>(), stride);

    ASSERT_EQ(offset.dimension_0(), 3u);
    ASSERT_EQ(offset.dimension_1(), 4u);
    ASSERT_EQ(offset.dimension_2(), 5u);

    ASSERT_EQ(offset.stride_0(), 4u);
    ASSERT_EQ(offset.stride_1(), 1u);
    ASSERT_EQ(offset.stride_2(), 12u);

    ASSERT_EQ(offset.span(), 60u);
    ASSERT_TRUE(offset.span_is_contiguous());

    Kokkos::Impl::ViewMapping<traits_t, void> v(
        Kokkos::Impl::ViewCtorProp<int*>(nullptr), stride);
  }

  {
    using V           = Kokkos::View<int**, Space>;
    using M           = typename V::host_mirror_type;
    using layout_type = typename Kokkos::View<int**, Space>::array_layout;

    constexpr size_t N0 = 10;
    constexpr size_t N1 = 11;

    V a("a", N0, N1);
    M b = Kokkos::create_mirror(a);
    M c = Kokkos::create_mirror_view(a);
    M d;

    for (size_t i0 = 0; i0 < N0; ++i0)
      for (size_t i1 = 0; i1 < N1; ++i1) {
        b(i0, i1) = 1 + i0 + i1 * N0;
      }

    Kokkos::deep_copy(a, b);
    Kokkos::deep_copy(c, a);

    for (size_t i0 = 0; i0 < N0; ++i0)
      for (size_t i1 = 0; i1 < N1; ++i1) {
        ASSERT_EQ(b(i0, i1), c(i0, i1));
      }

    Kokkos::resize(b, 5, 6);

    for (int i0 = 0; i0 < 5; ++i0)
      for (int i1 = 0; i1 < 6; ++i1) {
        int val = 1 + i0 + i1 * N0;
        ASSERT_EQ(b(i0, i1), c(i0, i1));
        ASSERT_EQ(b(i0, i1), val);
      }

    Kokkos::realloc(c, 5, 6);
    Kokkos::realloc(d, 5, 6);

    ASSERT_EQ(b.extent(0), 5u);
    ASSERT_EQ(b.extent(1), 6u);
    ASSERT_EQ(c.extent(0), 5u);
    ASSERT_EQ(c.extent(1), 6u);
    ASSERT_EQ(d.extent(0), 5u);
    ASSERT_EQ(d.extent(1), 6u);

    layout_type layout(7, 8);
    Kokkos::resize(b, layout);
    for (int i0 = 0; i0 < 7; ++i0)
      for (int i1 = 6; i1 < 8; ++i1) {
        b(i0, i1) = 1 + i0 + i1 * N0;
      }

    for (int i0 = 5; i0 < 7; ++i0)
      for (int i1 = 0; i1 < 8; ++i1) {
        b(i0, i1) = 1 + i0 + i1 * N0;
      }

    for (int i0 = 0; i0 < 7; ++i0)
      for (int i1 = 0; i1 < 8; ++i1) {
        int val = 1 + i0 + i1 * N0;
        ASSERT_EQ(b(i0, i1), val);
      }

    Kokkos::realloc(c, layout);
    Kokkos::realloc(d, layout);

    ASSERT_EQ(b.extent(0), 7u);
    ASSERT_EQ(b.extent(1), 8u);
    ASSERT_EQ(c.extent(0), 7u);
    ASSERT_EQ(c.extent(1), 8u);
    ASSERT_EQ(d.extent(0), 7u);
    ASSERT_EQ(d.extent(1), 8u);
  }

  {
    using V = Kokkos::View<int**, Kokkos::LayoutStride, Space>;
    using M = typename V::host_mirror_type;
    using layout_type =
        typename Kokkos::View<int**, Kokkos::LayoutStride, Space>::array_layout;

    constexpr size_t N0 = 10;
    constexpr size_t N1 = 11;

    const int dimensions[] = {N0, N1};
    const int order[]      = {1, 0};

    V a("a", Kokkos::LayoutStride::order_dimensions(2, order, dimensions));
    M b = Kokkos::create_mirror(a);
    M c = Kokkos::create_mirror_view(a);
    M d;

    for (size_t i0 = 0; i0 < N0; ++i0)
      for (size_t i1 = 0; i1 < N1; ++i1) {
        b(i0, i1) = 1 + i0 + i1 * N0;
      }

    Kokkos::deep_copy(a, b);
    Kokkos::deep_copy(c, a);

    for (size_t i0 = 0; i0 < N0; ++i0)
      for (size_t i1 = 0; i1 < N1; ++i1) {
        ASSERT_EQ(b(i0, i1), c(i0, i1));
      }

    const int dimensions2[] = {7, 8};
    const int order2[]      = {1, 0};
    layout_type layout = layout_type::order_dimensions(2, order2, dimensions2);
    Kokkos::resize(b, layout);

    for (int i0 = 0; i0 < 7; ++i0)
      for (int i1 = 0; i1 < 8; ++i1) {
        int val = 1 + i0 + i1 * N0;
        ASSERT_EQ(b(i0, i1), c(i0, i1));
        ASSERT_EQ(b(i0, i1), val);
      }

    Kokkos::realloc(c, layout);
    Kokkos::realloc(d, layout);

    ASSERT_EQ(b.extent(0), 7u);
    ASSERT_EQ(b.extent(1), 8u);
    ASSERT_EQ(c.extent(0), 7u);
    ASSERT_EQ(c.extent(1), 8u);
    ASSERT_EQ(d.extent(0), 7u);
    ASSERT_EQ(d.extent(1), 8u);
  }

  {
    using V = Kokkos::View<int*, Space>;
    using U = Kokkos::View<int*, Space, Kokkos::MemoryUnmanaged>;

    V a("a", 10);

    ASSERT_EQ(a.use_count(), 1);

    V b = a;

    ASSERT_EQ(a.use_count(), 2);
    ASSERT_EQ(b.use_count(), 2);

    {
      U c = b;  // 'c' is compile-time unmanaged.

      ASSERT_EQ(a.use_count(), 2);
      ASSERT_EQ(b.use_count(), 2);
#ifdef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
      ASSERT_EQ(c.use_count(), 2);
#else
      ASSERT_EQ(c.use_count(), 0);
#endif

      V d = c;  // 'd' is run-time unmanaged.

      ASSERT_EQ(a.use_count(), 2);
      ASSERT_EQ(b.use_count(), 2);
// FIXME: Legacy View is weird: it passes on use count even through compile time
// unmanaged transition
#ifdef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
      ASSERT_EQ(c.use_count(), 2);
      ASSERT_EQ(d.use_count(), 2);
#else
      ASSERT_EQ(c.use_count(), 0);
      ASSERT_EQ(d.use_count(), 0);
#endif
    }

    ASSERT_EQ(a.use_count(), 2);
    ASSERT_EQ(b.use_count(), 2);

    b = V();

    ASSERT_EQ(a.use_count(), 1);
    ASSERT_EQ(b.use_count(), 0);

// FIXME_NVCC For some reason, the use count is higher (but still constant) when
// using nvcc. Replacing the lambda with a functor doesn't show this behavior.
#if !(defined(KOKKOS_ENABLE_CUDA) && defined(KOKKOS_COMPILER_NVCC))
    using host_exec_space = typename Kokkos::Impl::HostMirror<
        typename Space::memory_space>::execution_space;

    int errors = 0;
    Kokkos::parallel_reduce(
        Kokkos::RangePolicy<host_exec_space>(0, 10),
        KOKKOS_LAMBDA(int, int& e) {
          // for parallel_reduce we copy the functor into a combined
          // functor-reducer object (with reference-counting on) before
          // constructing the ParallelReduce object (with reference-counting
          // turned off). When the parallel dispatch accepts a move for the
          // lambda, this count should become 2.

          if (a.use_count() != 3) ++e;
          V x = a;
          if (a.use_count() != 3) ++e;
          if (x.use_count() != 3) ++e;
        },
        errors);
    ASSERT_EQ(errors, 0);
#endif
  }
}

TEST(TEST_CATEGORY, view_mapping) { test_view_mapping<TEST_EXECSPACE>(); }
/*--------------------------------------------------------------------------*/

template <class ViewType>
struct TestViewMapOperator {
  static_assert(ViewType::reference_type_is_lvalue_reference,
                "Test only valid for lvalue reference type");

  ViewType v;

  template <size_t R>
  KOKKOS_INLINE_FUNCTION size_t extent_dim() const {
#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    return v.extent(R);
#else
    if constexpr (R < ViewType::rank) {
      return v.extent(R);
    } else {
      return 1;
    }
#endif
  }

  template <typename T>
  KOKKOS_INLINE_FUNCTION void test(size_t i0, int64_t& error_count) const {
    typename ViewType::value_type* const base_ptr =
        &v.access(0, 0, 0, 0, 0, 0, 0, 0);
    int64_t offset = 0;

    const size_t n1 = extent_dim<1>();
    const size_t n2 = extent_dim<2>();
    const size_t n3 = extent_dim<3>();
    const size_t n4 = extent_dim<4>();
    const size_t n5 = extent_dim<5>();
    const size_t n6 = extent_dim<6>();
    const size_t n7 = extent_dim<7>();

    if constexpr (std::is_same_v<T, Kokkos::LayoutRight>) {
      for (size_t i1 = 0; i1 < n1; ++i1)
        for (size_t i2 = 0; i2 < n2; ++i2)
          for (size_t i3 = 0; i3 < n3; ++i3)
            for (size_t i4 = 0; i4 < n4; ++i4)
              for (size_t i5 = 0; i5 < n5; ++i5)
                for (size_t i6 = 0; i6 < n6; ++i6)
                  for (size_t i7 = 0; i7 < n7; ++i7) {
                    const int64_t d =
                        &v.access(i0, i1, i2, i3, i4, i5, i6, i7) - base_ptr;
                    if (d < offset) ++error_count;
                    offset = d;
                  }
    } else {
      for (size_t i7 = 0; i7 < n7; ++i7)
        for (size_t i6 = 0; i6 < n6; ++i6)
          for (size_t i5 = 0; i5 < n5; ++i5)
            for (size_t i4 = 0; i4 < n4; ++i4)
              for (size_t i3 = 0; i3 < n3; ++i3)
                for (size_t i2 = 0; i2 < n2; ++i2)
                  for (size_t i1 = 0; i1 < n1; ++i1) {
                    const int64_t d =
                        &v.access(i0, i1, i2, i3, i4, i5, i6, i7) - base_ptr;
                    if (d < offset) ++error_count;
                    offset = d;
                  }
    }

    if (v.span() <= size_t(offset)) ++error_count;
  }

  KOKKOS_INLINE_FUNCTION
  void operator()(size_t i, int64_t& error_count) const {
    test<typename ViewType::array_layout>(i, error_count);
  }

  enum { N0 = 10 };
  enum { N1 = 9 };
  enum { N2 = 8 };
  enum { N3 = 7 };
  enum { N4 = 6 };
  enum { N5 = 5 };
  enum { N6 = 4 };
  enum { N7 = 3 };

  TestViewMapOperator() {
    const size_t dyn_rank = v.rank_dynamic;
    const std::string label("Test");
    switch (dyn_rank) {
      case 0: v = ViewType(label); break;
      case 1: v = ViewType(label, N0); break;
      case 2: v = ViewType(label, N0, N1); break;
      case 3: v = ViewType(label, N0, N1, N2); break;
      case 4: v = ViewType(label, N0, N1, N2, N3); break;
      case 5: v = ViewType(label, N0, N1, N2, N3, N4); break;
      case 6: v = ViewType(label, N0, N1, N2, N3, N4, N5); break;
      case 7: v = ViewType(label, N0, N1, N2, N3, N4, N5, N6); break;
      case 8:
      default: v = ViewType(label, N0, N1, N2, N3, N4, N5, N6, N7);
    }
  }

  void run() {
#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    ASSERT_EQ(
        v.extent(0),
        (size_t)(0 < ViewType::rank ? TestViewMapOperator<ViewType>::N0 : 1));
    ASSERT_EQ(
        v.extent(1),
        (size_t)(1 < ViewType::rank ? TestViewMapOperator<ViewType>::N1 : 1));
    ASSERT_EQ(
        v.extent(2),
        (size_t)(2 < ViewType::rank ? TestViewMapOperator<ViewType>::N2 : 1));
    ASSERT_EQ(
        v.extent(3),
        (size_t)(3 < ViewType::rank ? TestViewMapOperator<ViewType>::N3 : 1));
    ASSERT_EQ(
        v.extent(4),
        (size_t)(4 < ViewType::rank ? TestViewMapOperator<ViewType>::N4 : 1));
    ASSERT_EQ(
        v.extent(5),
        (size_t)(5 < ViewType::rank ? TestViewMapOperator<ViewType>::N5 : 1));
    ASSERT_EQ(
        v.extent(6),
        (size_t)(6 < ViewType::rank ? TestViewMapOperator<ViewType>::N6 : 1));
    ASSERT_EQ(
        v.extent(7),
        (size_t)(7 < ViewType::rank ? TestViewMapOperator<ViewType>::N7 : 1));

    ASSERT_LE(v.extent(0) * v.extent(1) * v.extent(2) * v.extent(3) *
                  v.extent(4) * v.extent(5) * v.extent(6) * v.extent(7),
              v.span());

#else
    if constexpr (0 < ViewType::rank) {
      ASSERT_EQ(v.extent(0), (size_t)TestViewMapOperator<ViewType>::N0);
    }
    if constexpr (1 < ViewType::rank) {
      ASSERT_EQ(v.extent(1), (size_t)TestViewMapOperator<ViewType>::N1);
    }
    if constexpr (2 < ViewType::rank) {
      ASSERT_EQ(v.extent(2), (size_t)TestViewMapOperator<ViewType>::N2);
    }
    if constexpr (3 < ViewType::rank) {
      ASSERT_EQ(v.extent(3), (size_t)TestViewMapOperator<ViewType>::N3);
    }
    if constexpr (4 < ViewType::rank) {
      ASSERT_EQ(v.extent(4), (size_t)TestViewMapOperator<ViewType>::N4);
    }
    if constexpr (5 < ViewType::rank) {
      ASSERT_EQ(v.extent(5), (size_t)TestViewMapOperator<ViewType>::N5);
    }
    if constexpr (6 < ViewType::rank) {
      ASSERT_EQ(v.extent(6), (size_t)TestViewMapOperator<ViewType>::N6);
    }
    if constexpr (7 < ViewType::rank) {
      ASSERT_EQ(v.extent(7), (size_t)TestViewMapOperator<ViewType>::N7);
    }

#endif
    size_t extent;
#ifdef KOKKOS_ENABLE_DEPRECATED_CODE_5
    extent = v.extent(0);
#else
    // Make sure the view is at least of size 1
    extent = 1;
    if constexpr (ViewType::rank > 0) extent = v.extent(0);
#endif
    int64_t error_count = 0;
    Kokkos::RangePolicy<typename ViewType::execution_space> range(0, extent);
    Kokkos::parallel_reduce(range, *this, error_count);
    ASSERT_EQ(0, error_count);
  }
};

template <class Space>
void test_view_mapping_operator() {
  using ExecSpace = typename Space::execution_space;

  {
    TestViewMapOperator<Kokkos::View<int, Kokkos::LayoutLeft, ExecSpace> > f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int*, Kokkos::LayoutLeft, ExecSpace> > f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int**, Kokkos::LayoutLeft, ExecSpace> > f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int***, Kokkos::LayoutLeft, ExecSpace> > f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int****, Kokkos::LayoutLeft, ExecSpace> >
        f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int*****, Kokkos::LayoutLeft, ExecSpace> >
        f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int******, Kokkos::LayoutLeft, ExecSpace> >
        f;
    f.run();
  }
  {
    TestViewMapOperator<
        Kokkos::View<int*******, Kokkos::LayoutLeft, ExecSpace> >
        f;
    f.run();
  }

  {
    TestViewMapOperator<Kokkos::View<int, Kokkos::LayoutRight, ExecSpace> > f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int*, Kokkos::LayoutRight, ExecSpace> > f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int**, Kokkos::LayoutRight, ExecSpace> > f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int***, Kokkos::LayoutRight, ExecSpace> >
        f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int****, Kokkos::LayoutRight, ExecSpace> >
        f;
    f.run();
  }
  {
    TestViewMapOperator<Kokkos::View<int*****, Kokkos::LayoutRight, ExecSpace> >
        f;
    f.run();
  }
  {
    TestViewMapOperator<
        Kokkos::View<int******, Kokkos::LayoutRight, ExecSpace> >
        f;
    f.run();
  }
  {
    TestViewMapOperator<
        Kokkos::View<int*******, Kokkos::LayoutRight, ExecSpace> >
        f;
    f.run();
  }
}

TEST(TEST_CATEGORY, view_mapping_operator) {
  test_view_mapping_operator<TEST_EXECSPACE>();
}

TEST(TEST_CATEGORY, static_extent) {
  using T = Kokkos::View<double* [2][3]>;
  ASSERT_EQ(T::static_extent(1), 2u);
  ASSERT_EQ(T::static_extent(2), 3u);
}

}  // namespace Test
