Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Use autodiff in templated network rate evaluation #1614

Merged
merged 16 commits into from
Jul 26, 2024
Merged

Use autodiff in templated network rate evaluation #1614

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
ec07beb
Template tf_t and rate_t on dual_t
yut23 Jul 17, 2024
8cc4013
Update rhs.H to use templated rate_t
yut23 Jul 17, 2024
e866293
Update aprox_rates.H and test_aprox_rates
yut23 Jul 17, 2024
98c70e1
Update networks that use aprox_rates.H
yut23 Jul 17, 2024
53ca9a7
Update rprox
yut23 Jul 17, 2024
0d89086
Update the remaining templated networks
yut23 Jul 17, 2024
41aef6f
Remove some debugging stuff from aprox21
yut23 Jul 17, 2024
d4173de
Update test_rhs benchmark for powerlaw
yut23 Jul 17, 2024
ec2c297
Update burn_cell benchmark for ignition_chamulak
yut23 Jul 17, 2024
3fdf278
Merge remote-tracking branch 'origin/development' into autodiff_aprox…
yut23 Jul 17, 2024
d37f4bc
Merge branch 'development' into autodiff_aprox_rates
zingale Jul 18, 2024
541fd6b
Merge branch 'development' into autodiff_aprox_rates
zingale Jul 22, 2024
21f9e71
Revert some optimizations that were changing results slightly
yut23 Jul 23, 2024
350928a
Update ignition_chamulak benchmark
yut23 Jul 23, 2024
3afa5ef
Replace dual_t with number_t in the newly-converted code
yut23 Jul 23, 2024
1bc3dd4
Merge branch 'development' into autodiff_aprox_rates
yut23 Jul 26, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions interfaces/rhs_type.H
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -95,12 +95,13 @@ struct rate_tab_t
}
};

// number_t is currently only used in the screening routines
template <typename number_t = amrex::Real>
// number_t is currently used in the screening routines and templated network
// rate evaluation
template <typename number_t>
struct rhs_state_t
{
amrex::Real rho;
tf_t tf;
tf_t<number_t> tf;
rate_tab_t tab;
#ifdef SCREENING
plasma_state_t<number_t> pstate;
Expand All @@ -110,12 +111,11 @@ struct rhs_state_t
amrex::Array1D<amrex::Real, 1, NumSpec> y;
};

template <typename number_t>
struct rate_t
{
amrex::Real fr;
amrex::Real rr;
amrex::Real frdt;
amrex::Real rrdt;
number_t fr;
number_t rr;
};

} // namespace RHS
Expand Down
142 changes: 73 additions & 69 deletions interfaces/tfactors.H
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,76 +2,80 @@
#define TFACTORS_H

#include <AMReX.H>
#include <AMReX_REAL.H>
#include <microphysics_autodiff.H>
#include <cmath>

using namespace amrex::literals;

template <typename number_t>
struct tf_t {
amrex::Real temp;
amrex::Real t9;
amrex::Real t92;
amrex::Real t93;
// amrex::Real t94;
amrex::Real t95;
// amrex::Real t96;
amrex::Real t912;
amrex::Real t932;
amrex::Real t952;
amrex::Real t972;
amrex::Real t913;
amrex::Real t923;
amrex::Real t943;
amrex::Real t953;
// amrex::Real t973;
// amrex::Real t9113;
// amrex::Real t914;
// amrex::Real t934;
// amrex::Real t954;
// amrex::Real t974;
// amrex::Real t915;
// amrex::Real t935;
// amrex::Real t945;
// amrex::Real t965;
// amrex::Real t917;
// amrex::Real t927;
// amrex::Real t947;
// amrex::Real t918;
// amrex::Real t938;
// amrex::Real t958;
amrex::Real t9i;
amrex::Real t9i2;
// amrex::Real t9i3;
amrex::Real t9i12;
amrex::Real t9i32;
// amrex::Real t9i52;
// amrex::Real t9i72;
amrex::Real t9i13;
amrex::Real t9i23;
amrex::Real t9i43;
amrex::Real t9i53;
// amrex::Real t9i14;
// amrex::Real t9i34;
// amrex::Real t9i54;
// amrex::Real t9i15;
// amrex::Real t9i35;
// amrex::Real t9i45;
// amrex::Real t9i65;
// amrex::Real t9i17;
// amrex::Real t9i27;
// amrex::Real t9i47;
// amrex::Real t9i18;
// amrex::Real t9i38;
// amrex::Real t9i58;
// amrex::Real t916;
// amrex::Real t976;
// amrex::Real t9i76;
amrex::Real lnt9;
number_t temp;
number_t t9;
number_t t92;
number_t t93;
// number_t t94;
number_t t95;
// number_t t96;
number_t t912;
number_t t932;
number_t t952;
number_t t972;
number_t t913;
number_t t923;
number_t t943;
number_t t953;
// number_t t973;
// number_t t9113;
// number_t t914;
// number_t t934;
// number_t t954;
// number_t t974;
// number_t t915;
// number_t t935;
// number_t t945;
// number_t t965;
// number_t t917;
// number_t t927;
// number_t t947;
// number_t t918;
// number_t t938;
// number_t t958;
number_t t9i;
number_t t9i2;
// number_t t9i3;
number_t t9i12;
number_t t9i32;
// number_t t9i52;
// number_t t9i72;
number_t t9i13;
number_t t9i23;
number_t t9i43;
number_t t9i53;
// number_t t9i14;
// number_t t9i34;
// number_t t9i54;
// number_t t9i15;
// number_t t9i35;
// number_t t9i45;
// number_t t9i65;
// number_t t9i17;
// number_t t9i27;
// number_t t9i47;
// number_t t9i18;
// number_t t9i38;
// number_t t9i58;
// number_t t916;
// number_t t976;
// number_t t9i76;
number_t lnt9;
};

template <typename number_t>
AMREX_GPU_HOST_DEVICE inline
tf_t get_tfactors(amrex::Real temp)
tf_t<number_t> get_tfactors(number_t temp)
{
tf_t tf;
tf_t<number_t> tf;

tf.temp = temp;

Expand All @@ -83,38 +87,38 @@ tf_t get_tfactors(amrex::Real temp)
tf.t95 = tf.t92*tf.t93;
// tf.t96 = tf.t9*tf.t95;

tf.t912 = std::sqrt(tf.t9);
tf.t912 = admath::sqrt(tf.t9);
tf.t932 = tf.t9*tf.t912;
tf.t952 = tf.t9*tf.t932;
// tf.t972 = tf.t9*tf.t952;
tf.t972 = tf.t92*tf.t932;

tf.t913 = std::cbrt(tf.t9);
tf.t913 = admath::cbrt(tf.t9);
tf.t923 = tf.t913*tf.t913;
tf.t943 = tf.t9*tf.t913;
tf.t953 = tf.t9*tf.t923;
// tf.t973 = tf.t953*tf.t923;
// tf.t9113 = tf.t973*tf.t943;

// tf.t914 = std::pow(tf.t9, 0.25e0_rt);
// tf.t914 = admath::pow(tf.t9, 0.25e0_rt);
// tf.t934 = tf.t914*tf.t914*tf.t914;
// tf.t954 = tf.t9*tf.t914;
// tf.t974 = tf.t9*tf.t934;

// tf.t915 = std::pow(tf.t9, 0.2_rt);
// tf.t915 = admath::pow(tf.t9, 0.2_rt);
// tf.t935 = tf.t915*tf.t915*tf.t915;
// tf.t945 = tf.t915 * tf.t935;
// tf.t965 = tf.t9 * tf.t915;

// tf.t916 = std::pow(tf.t9, 1.0_rt/6.0_rt);
// tf.t916 = admath::pow(tf.t9, 1.0_rt/6.0_rt);
// tf.t976 = tf.t9 * tf.t916;
// tf.t9i76 = 1.0e0_rt/tf.t976;

// tf.t917 = std::pow(tf.t9, 1.0_rt/7.0_rt);
// tf.t917 = admath::pow(tf.t9, 1.0_rt/7.0_rt);
// tf.t927 = tf.t917*tf.t917;
// tf.t947 = tf.t927*tf.t927;

// tf.t918 = std::sqrt(tf.t914);
// tf.t918 = admath::sqrt(tf.t914);
// tf.t938 = tf.t918*tf.t918*tf.t918;
// tf.t958 = tf.t938*tf.t918*tf.t918;

Expand Down Expand Up @@ -149,7 +153,7 @@ tf_t get_tfactors(amrex::Real temp)
// tf.t9i38 = tf.t9i18*tf.t9i18*tf.t9i18;
// tf.t9i58 = tf.t9i38*tf.t9i18*tf.t9i18;

tf.lnt9 = std::log(tf.t9);
tf.lnt9 = admath::log(tf.t9);

return tf;
}
Expand Down
Loading
Loading