diff --git a/integration/nse_update_sdc.H b/integration/nse_update_sdc.H index 2b6daae07..7d6736d38 100644 --- a/integration/nse_update_sdc.H +++ b/integration/nse_update_sdc.H @@ -33,7 +33,7 @@ using namespace amrex::literals; /// on input, *_source are the reactive sources at time t0 and on output /// they are the sources at time t0+dt /// -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void nse_derivs(const amrex::Real rho0, const amrex::Real rhoe0, const amrex::Real *rhoaux0, const amrex::Real dt, const amrex::Real *ydot_a, amrex::Real& drhoedt, amrex::Real* drhoauxdt, const amrex::Real T_fixed) { @@ -168,7 +168,7 @@ void nse_derivs(const amrex::Real rho0, const amrex::Real rhoe0, const amrex::Re /// this version works with the tabulated NSE and requires AUX_THERMO /// template -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void sdc_nse_burn(BurnT& state, const amrex::Real dt) { using namespace AuxZero; @@ -307,7 +307,7 @@ void sdc_nse_burn(BurnT& state, const amrex::Real dt) { /// /// This computes drhoedt and drhoyedt_weak for a given (rho, rho e, and rho Ye) /// -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void nse_derivs(const amrex::Real rho0, const amrex::Real rhoe0, const amrex::Real rhoYe0, amrex::Real &T0, amrex::Real &mu_p, amrex::Real &mu_n, @@ -415,7 +415,7 @@ void nse_derivs(const amrex::Real rho0, const amrex::Real rhoe0, /// template -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void sdc_nse_burn(BurnT& state, const amrex::Real dt) { state.success = true; diff --git a/nse_solver/nse_check.H b/nse_solver/nse_check.H index d7f0b016d..42fd7a8cb 100644 --- a/nse_solver/nse_check.H +++ b/nse_solver/nse_check.H @@ -14,7 +14,7 @@ #include -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void check_nse_molar(const amrex::Array1D& Y, const amrex::Array1D& Y_nse, bool& nse_check) { @@ -65,7 +65,7 @@ void check_nse_molar(const amrex::Array1D& Y, } -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE int get_root_index(const int nuc_idx, amrex::Array1D& group_idx) { // Returns the root index of the nuclei given the nuclei index [1, NumSpec] // Also updates all parent indices to point to the root index in group_idx @@ -95,7 +95,7 @@ int get_root_index(const int nuc_idx, amrex::Array1D& group_idx } -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void nse_union(const int nuc_idx_a, const int nuc_idx_b, amrex::Array1D& group_idx, amrex::Array1D& group_size) { @@ -126,7 +126,7 @@ void nse_union(const int nuc_idx_a, const int nuc_idx_b, } -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE bool in_single_group(amrex::Array1D& group_idx) { // This function checks whether all isotopes are either in the LIG group @@ -146,7 +146,7 @@ bool in_single_group(amrex::Array1D& group_idx) { template -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void fill_reaction_timescale(amrex::Array1D& reaction_timescales, const int current_rate_pair, const amrex::Real rho, const amrex::Array1D& Y, @@ -238,7 +238,7 @@ void fill_reaction_timescale(amrex::Array1D& r } -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void fill_merge_indices(int& merge_idx1, int& merge_idx2, const int current_rate_pair, amrex::Array1D& group_idx) { @@ -288,7 +288,7 @@ void fill_merge_indices(int& merge_idx1, int& merge_idx2, } -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE void nse_grouping(amrex::Array1D& group_idx, const amrex::Real rho, const amrex::Array1D& Y, const amrex::Array1D& screened_rates, @@ -374,7 +374,7 @@ void nse_grouping(amrex::Array1D& group_idx, const amrex::Real } -AMREX_GPU_HOST_DEVICE AMREX_INLINE +AMREX_GPU_DEVICE AMREX_INLINE bool in_nse(burn_t& current_state, bool skip_molar_check=false) { // This function returns the boolean that tells whether we're in nse or not