Skip to content

Commit

Permalink
add test for double ^ adouble
Browse files Browse the repository at this point in the history
  • Loading branch information
@TimSiebert1
TimSiebert1 committed Jan 9, 2025
1 parent e6cdb5f commit 3471a31
Show file tree
Hide file tree
Showing 2 changed files with 149 additions and 0 deletions.
74 changes: 74 additions & 0 deletions ADOL-C/boost-test/uni5_for/hov_forward.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2795,4 +2795,78 @@ BOOST_AUTO_TEST_CASE(PowOperator_HOV_Forward_2) {
myfree3(X);
myfree3(Y);
}

BOOST_AUTO_TEST_CASE(PowOperator_HOV_Forward_3) {
const int16_t tag = 0;
const size_t dim_out = 1;
const size_t dim_in = 1;
const size_t degree = 3;
const size_t num_dirs = 2;
std::vector<double> in{2.1};
std::vector<adouble> indep(dim_in);
std::vector<double> out(dim_out);

// x^y
trace_on(tag);
indep[0] <<= in[0];
adouble dep = pow(1.5, indep[0]);
dep >>= out[0];
trace_off();

double ***X = myalloc3(dim_in, num_dirs, degree);
double ***Y = myalloc3(dim_out, num_dirs, degree);

X[0][0][0] = 1.0;
X[0][0][1] = -1.0;
X[0][0][2] = 1.1;

X[0][1][0] = 2.0;
X[0][1][1] = -2.1;
X[0][1][2] = -2.0;

std::vector<double> test_in{2.1};

// x^y
double test_out = std::pow(1.5, test_in[0]);

hov_forward(tag, dim_out, dim_in, degree, num_dirs, test_in.data(), X,
out.data(), Y);

BOOST_TEST(out[0] == test_out, tt::tolerance(tol));

const double log_x = std::log(1.5);
// first derivative
BOOST_TEST(Y[0][0][0] == std::pow(1.5, test_in[0]) * log_x * X[0][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][0] == std::pow(1.5, test_in[0]) * log_x * X[0][1][0],
tt::tolerance(tol));

// second derivative
BOOST_TEST(Y[0][0][1] == std::pow(1.5, test_in[0]) * log_x * X[0][0][1] +
1.0 / 2.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * X[0][0][0] * X[0][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][1] == std::pow(1.5, test_in[0]) * log_x * X[0][1][1] +
1.0 / 2.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * X[0][1][0] * X[0][1][0],
tt::tolerance(tol));

// third derivative
BOOST_TEST(Y[0][0][2] == std::pow(1.5, test_in[0]) * log_x * X[0][0][2] +
std::pow(1.5, test_in[0]) * log_x * log_x *
X[0][0][0] * X[0][0][1] +
1.0 / 6.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * log_x * X[0][0][0] * X[0][0][0] *
X[0][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][2] == std::pow(1.5, test_in[0]) * log_x * X[0][1][2] +
std::pow(1.5, test_in[0]) * log_x * log_x *
X[0][1][0] * X[0][1][1] +
1.0 / 6.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * log_x * X[0][1][0] * X[0][1][0] *
X[0][1][0],
tt::tolerance(tol));
myfree3(X);
myfree3(Y);
}
BOOST_AUTO_TEST_SUITE_END()
75 changes: 75 additions & 0 deletions ADOL-C/boost-test/uni5_for/hov_wk_forward.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2824,4 +2824,79 @@ BOOST_AUTO_TEST_CASE(PowOperator_HOV_Forward_2) {
myfree3(Y);
}

BOOST_AUTO_TEST_CASE(PowOperator_HOV_Forward_3) {
const int16_t tag = 0;
const size_t dim_out = 1;
const size_t dim_in = 1;
const size_t degree = 3;
const size_t num_dirs = 2;
const short keep = 1;
std::vector<double> in{2.1};
std::vector<adouble> indep(dim_in);
std::vector<double> out(dim_out);

// x^y
trace_on(tag);
indep[0] <<= in[0];
adouble dep = pow(1.5, indep[0]);
dep >>= out[0];
trace_off();

double ***X = myalloc3(dim_in, num_dirs, degree);
double ***Y = myalloc3(dim_out, num_dirs, degree);

X[0][0][0] = 1.0;
X[0][0][1] = -1.0;
X[0][0][2] = 1.1;

X[0][1][0] = 2.0;
X[0][1][1] = -2.1;
X[0][1][2] = -2.0;

std::vector<double> test_in{2.1};

// x^y
double test_out = std::pow(1.5, test_in[0]);

hov_wk_forward(tag, dim_out, dim_in, degree, keep, num_dirs, test_in.data(),
X, out.data(), Y);

BOOST_TEST(out[0] == test_out, tt::tolerance(tol));

const double log_x = std::log(1.5);
// first derivative
BOOST_TEST(Y[0][0][0] == std::pow(1.5, test_in[0]) * log_x * X[0][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][0] == std::pow(1.5, test_in[0]) * log_x * X[0][1][0],
tt::tolerance(tol));

// second derivative
BOOST_TEST(Y[0][0][1] == std::pow(1.5, test_in[0]) * log_x * X[0][0][1] +
1.0 / 2.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * X[0][0][0] * X[0][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][1] == std::pow(1.5, test_in[0]) * log_x * X[0][1][1] +
1.0 / 2.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * X[0][1][0] * X[0][1][0],
tt::tolerance(tol));

// third derivative
BOOST_TEST(Y[0][0][2] == std::pow(1.5, test_in[0]) * log_x * X[0][0][2] +
std::pow(1.5, test_in[0]) * log_x * log_x *
X[0][0][0] * X[0][0][1] +
1.0 / 6.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * log_x * X[0][0][0] * X[0][0][0] *
X[0][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][2] == std::pow(1.5, test_in[0]) * log_x * X[0][1][2] +
std::pow(1.5, test_in[0]) * log_x * log_x *
X[0][1][0] * X[0][1][1] +
1.0 / 6.0 * std::pow(1.5, test_in[0]) * log_x *
log_x * log_x * X[0][1][0] * X[0][1][0] *
X[0][1][0],
tt::tolerance(tol));
myfree3(X);
myfree3(Y);
}

BOOST_AUTO_TEST_SUITE_END()

0 comments on commit 3471a31

Please sign in to comment.