unfused_push_algorithm.hpp

Go to the documentation of this file.

#ifndef UNFUSED_PUSH_ALGORITHM_HPP
#define UNFUSED_PUSH_ALGORITHM_HPP
#include "globals.hpp"
#include "magnetic_field.hpp"
#include "grid.hpp"
#include "particles.hpp"
#include "species.hpp"
#include "perturbed_B_field.hpp"

/* This struct contains views that store intermediate values of the electron push so that the kernel can be split into
 * smaller pieces
 */
template<class Device>
struct UnfusedPushTmpViews{
    // Storage views used to split the electron push kernel
    Cabana::AoSoA<PhaseDataTypes,Device,VEC_LEN> phase0;
    Cabana::AoSoA<PhaseDataTypes,Device,VEC_LEN> dy;
    Cabana::AoSoA<PhaseDataTypes,Device,VEC_LEN> dy_sum;
    View<int*,CLayout,Device> current_tr;

    UnfusedPushTmpViews(){}

    UnfusedPushTmpViews(int n)
      : phase0("phase0", n),
        dy("dy", n),
        dy_sum("dy_sum", n),
        current_tr(NoInit("current_tr"), n) {}

    void resize(int n){
        phase0.resize(n);
        dy.resize(n);
        dy_sum.resize(n);
        Kokkos::resize(current_tr, n);

        // Set current_tr to -1 - this might not be necessary
        Kokkos::deep_copy(current_tr, -1);
    }
};

void unfused_electron_push_kernel(Species<DeviceType> &species, const PushControls &push_controls, const DomainDecomposition<DeviceType>& pol_decomp,
        const Grid<DeviceType> &grid, const MagneticField<DeviceType> &magnetic_field, const GridFieldPack<DeviceType,PIT_GLOBAL> &gfpack,
        const GridFieldPack<DeviceType,PIT_GLOBAL> &gfpack_near,
        const PerturbedBField<DeviceType> &perturbed_B_field, SortViews<DeviceType>& sort_views,
        const Sheath<DeviceType> &sheath, const Neutrals<DeviceType> &neutrals,
        const HeatDiagnostics<DeviceType> &heat_diag, UnfusedPushTmpViews<DeviceType>& tmp, int ncycles_per_launch, double dt);

void local_shift(const PushControls &push_controls, Species<DeviceType> &species, SortViews<DeviceType>& sort_views, const Grid<DeviceType> &grid, const MagneticField<DeviceType> &magnetic_field, const DomainDecomposition<DeviceType> &pol_decomp, UnfusedPushTmpViews<DeviceType>& unfused_push_tmp_views, bool send_phase0, bool use_near_field);

#endif