File: TestViewSpaceAssign.hpp

package info (click to toggle)
kokkos 5.0.1-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 15,140 kB
  • sloc: cpp: 225,293; sh: 1,250; python: 78; makefile: 16; fortran: 4; ansic: 2
file content (39 lines) | stat: -rw-r--r-- 1,228 bytes parent folder | download
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
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project

#include <gtest/gtest.h>

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

namespace Test {

template <typename SpaceDst, typename SpaceSrc>
void view_space_assign() {
  Kokkos::View<double*, SpaceDst> a = Kokkos::View<double*, SpaceSrc>("a", 1);

  Kokkos::View<double*, Kokkos::LayoutLeft, SpaceDst> b =
      Kokkos::View<double*, Kokkos::LayoutLeft, SpaceSrc>("b", 1);

  Kokkos::View<double*, Kokkos::LayoutRight, SpaceDst> c =
      Kokkos::View<double*, Kokkos::LayoutRight, SpaceSrc>("c", 1);

  Kokkos::View<double*, SpaceDst, Kokkos::MemoryRandomAccess> d =
      Kokkos::View<double*, SpaceSrc>("d", 1);

  Kokkos::View<double*, Kokkos::LayoutLeft, SpaceDst,
               Kokkos::MemoryRandomAccess>
      e = Kokkos::View<double*, Kokkos::LayoutLeft, SpaceSrc>("e", 1);

  // Rank-one layout can assign:
  Kokkos::View<double*, Kokkos::LayoutRight, SpaceDst> f =
      Kokkos::View<double*, Kokkos::LayoutLeft, SpaceSrc>("f", 1);
}

}  // namespace Test