host_state.hpp

Go to the documentation of this file.

#pragma once
 
#include "state/state.hpp"
 
namespace kynema::interfaces {
 
template <typename DeviceType>
struct HostState {
    template <typename ValueType>
    using HostView = typename Kokkos::View<ValueType, DeviceType>::HostMirror;
 
    HostView<double* [7]> x;
 
    HostView<double* [7]> q;
 
    HostView<double* [6]> v;
 
    HostView<double* [6]> vd;
 
    HostView<double* [6]> f;
 
    explicit HostState(const State<DeviceType>& state)
        : x(Kokkos::create_mirror_view(Kokkos::WithoutInitializing, state.x)),
          q(Kokkos::create_mirror_view(Kokkos::WithoutInitializing, state.q)),
          v(Kokkos::create_mirror_view(Kokkos::WithoutInitializing, state.v)),
          vd(Kokkos::create_mirror_view(Kokkos::WithoutInitializing, state.vd)),
          f(Kokkos::create_mirror_view(Kokkos::WithoutInitializing, state.f)) {}
 
    void CopyFromState(const State<DeviceType>& state) const {
        Kokkos::deep_copy(this->x, state.x);
        Kokkos::deep_copy(this->q, state.q);
        Kokkos::deep_copy(this->v, state.v);
        Kokkos::deep_copy(this->vd, state.vd);
    }
};
 
}  // namespace kynema::interfaces