update_system_variables_masses.hpp

Go to the documentation of this file.

#pragma once
 
#include <Kokkos_Core.hpp>
#include <Kokkos_Profiling_ScopedRegion.hpp>
 
#include "elements/masses/masses.hpp"
#include "state/state.hpp"
#include "step_parameters.hpp"
#include "system/masses/calculate_quadrature_point_values.hpp"
 
namespace kynema::step {
 
template <typename DeviceType>
inline void UpdateSystemVariablesMasses(
    const StepParameters& parameters, const Masses<DeviceType>& masses, State<DeviceType>& state
) {
    auto region = Kokkos::Profiling::ScopedRegion("Update System Variables Masses");
 
    using RangePolicy = Kokkos::RangePolicy<typename DeviceType::execution_space>;
    auto range_policy = RangePolicy(0, masses.num_elems);
 
    Kokkos::parallel_for(
        "masses::CalculateQuadraturePointValues", range_policy,
        masses::CalculateQuadraturePointValues<DeviceType>{
            parameters.beta_prime, parameters.gamma_prime, state.q, state.v, state.vd, state.tangent,
            masses.state_indices, masses.gravity, masses.qp_Mstar, masses.node_x0,
            masses.residual_vector_terms, masses.system_matrix_terms
        }
    );
}
 
}  // namespace kynema::step