my_subview.hpp

Go to the documentation of this file.

#ifndef MY_SUBVIEW_HPP
#define MY_SUBVIEW_HPP

#include "space_settings.hpp"

// 4D to 1D
template<class Device, typename T>
Kokkos::View<T*,Kokkos::LayoutRight,Device> my_subview(const Kokkos::View<T****,Kokkos::LayoutRight,Device>& view, int i, int j, int k){
    return Kokkos::View<T*,Kokkos::LayoutRight,Device>(view, i, j, k, Kokkos::ALL());
}

// 3D to 1D
template<class Device, typename T>
Kokkos::View<T*,Kokkos::LayoutRight,Device> my_subview(const Kokkos::View<T***,Kokkos::LayoutRight,Device>& view, int i, int j){
    return Kokkos::View<T*,Kokkos::LayoutRight,Device>(view, i, j, Kokkos::ALL());
}

// 2D to 1D
template<class Device, typename T>
Kokkos::View<T*,Kokkos::LayoutRight,Device> my_subview(const Kokkos::View<T**,Kokkos::LayoutRight,Device>& view, int i){
    return Kokkos::View<T*,Kokkos::LayoutRight,Device>(view, i, Kokkos::ALL());
}

// 4D to 2D
template<class Device, typename T>
Kokkos::View<T**,Kokkos::LayoutRight,Device> my_subview(const Kokkos::View<T****,Kokkos::LayoutRight,Device>& view, int i, int j){
    return Kokkos::View<T**,Kokkos::LayoutRight,Device>(view, i, j, Kokkos::ALL(), Kokkos::ALL());
}

// 3D to 2D
template<class Device, typename T>
Kokkos::View<T**,Kokkos::LayoutRight,Device> my_subview(const Kokkos::View<T***,Kokkos::LayoutRight,Device>& view, int i){
    return Kokkos::View<T**,Kokkos::LayoutRight,Device>(view, i, Kokkos::ALL(), Kokkos::ALL());
}

// 4D to 3D
template<class Device, typename T>
Kokkos::View<T***,Kokkos::LayoutRight,Device> my_subview(const Kokkos::View<T****,Kokkos::LayoutRight,Device>& view, int i){
    return Kokkos::View<T***,Kokkos::LayoutRight,Device>(view, i, Kokkos::ALL(), Kokkos::ALL(), Kokkos::ALL());
}

// Unmanaged views
// 3D to 1D
template<class Device, typename T>
Kokkos::View<T*,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>> my_subview(const Kokkos::View<T***,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>& view, int i, int j){
    return Kokkos::View<T*,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>(view, i, j, Kokkos::ALL());
}

// 2D to 1D
template<class Device, typename T>
Kokkos::View<T*,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>> my_subview(const Kokkos::View<T**,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>& view, int i){
    return Kokkos::View<T*,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>(view, i, Kokkos::ALL());
}

// 4D to 2D
template<class Device, typename T>
Kokkos::View<T**,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>> my_subview(const Kokkos::View<T****,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>& view, int i, int j){
    return Kokkos::View<T**,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>(view, i, j, Kokkos::ALL(), Kokkos::ALL());
}

// 3D to 2D
template<class Device, typename T>
Kokkos::View<T**,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>> my_subview(const Kokkos::View<T***,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>& view, int i){
    return Kokkos::View<T**,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>(view, i, Kokkos::ALL(), Kokkos::ALL());
}

// 4D to 3D
template<class Device, typename T>
Kokkos::View<T***,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>> my_subview(const Kokkos::View<T****,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>& view, int i){
    return Kokkos::View<T***,Kokkos::LayoutRight,Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>(view, i, Kokkos::ALL(), Kokkos::ALL(), Kokkos::ALL());
}

#endif