Skip to content

Commit

Permalink
add tests for hov_forward with max
Browse files Browse the repository at this point in the history
  • Loading branch information
@TimSiebert1
TimSiebert1 committed Dec 11, 2024
1 parent 6c6cec3 commit cab7822
Showing 1 changed file with 161 additions and 1 deletion.
162 changes: 161 additions & 1 deletion ADOL-C/boost-test/uni5_for.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ BOOST_AUTO_TEST_CASE(Fmaxoperator_ZOS_PL_Forward) {
const int dim_out = 1;
const int dim_in = 3;

std::array<double, dim_in> in = {-2.0, 0.0, 1.5};
std::array<double, dim_in> in{-2.0, 0.0, 1.5};
std::array<adouble, dim_in> indep;
double out[] = {0.0};

Expand Down Expand Up @@ -82,4 +82,164 @@ BOOST_AUTO_TEST_CASE(Fmaxoperator_ZOS_PL_Forward) {

removeTape(tag, ADOLC_REMOVE_COMPLETELY);
}

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

trace_on(tag);
std::for_each(in.begin(), in.end(), [&, i = 0](int value) mutable {
indep[i] <<= in[i];
i++;
});

// max(x^2, y^3)
adouble dep = fmax(pow(indep[0], 2), pow(indep[1], 3));

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[1][0][0] = 1.9;

X[0][1][0] = 2.0;
X[1][1][0] = 3.0;

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

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

X[0][1][1] = -2.0;
X[1][1][1] = -3.0;

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

/****************************
TEST A < B
*****************************/
std::vector<double> test_in{4.0, 3.2};
// max(x^2, y^3)
double test_out = std::fmax(std::pow(test_in[0], 2), std::pow(test_in[1], 3));

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));
BOOST_TEST(Y[0][0][0] == 3 * std::pow(test_in[1], 2) * X[1][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][0] == 3 * std::pow(test_in[1], 2) * X[1][1][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][2][0] == 3 * std::pow(test_in[1], 2) * X[1][2][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][0][1] ==
1.0 / 2.0 * 6.0 * test_in[1] * X[1][0][0] * X[1][0][0] +
3.0 * std::pow(test_in[1], 2.0) * X[1][0][1],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][1] ==
1.0 / 2.0 * 6.0 * test_in[1] * X[1][1][0] * X[1][1][0] +
3.0 * std::pow(test_in[1], 2.0) * X[1][1][1],
tt::tolerance(tol));
BOOST_TEST(Y[0][2][1] ==
1.0 / 2.0 * 6.0 * test_in[1] * X[1][2][0] * X[1][2][0] +
3.0 * std::pow(test_in[1], 2.0) * X[1][2][1],
tt::tolerance(tol));

/**************************
TEST A > B
***************************/

// test_in[0] = 4.0
test_in[1] = 1.0;
// max(x^2, y^3)
test_out = std::fmax(std::pow(test_in[0], 2), std::pow(test_in[1], 3));
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));
BOOST_TEST(Y[0][0][0] == 2 * test_in[0] * X[0][0][0], tt::tolerance(tol));
BOOST_TEST(Y[0][1][0] == 2 * test_in[0] * X[0][1][0], tt::tolerance(tol));
BOOST_TEST(Y[0][2][0] == 2 * test_in[0] * X[0][2][0], tt::tolerance(tol));
BOOST_TEST(Y[0][0][1] == 1.0 / 2.0 * 2.0 * X[0][0][0] * X[0][0][0] +
2.0 * test_in[0] * X[0][0][1],
tt::tolerance(tol));
BOOST_TEST(Y[0][1][1] == 1.0 / 2.0 * 2.0 * X[0][1][0] * X[0][1][0] +
2.0 * test_in[0] * X[0][1][1],
tt::tolerance(tol));
BOOST_TEST(Y[0][2][1] == 1.0 / 2.0 * 2.0 * X[0][2][0] * X[0][2][0] +
2.0 * test_in[0] * X[0][2][1],
tt::tolerance(tol));

/**************************
TEST A == B
***************************/

test_in[0] = 1.0;
test_in[1] = 1.0;

// max(x^2, y^3)
test_out = std::fmax(std::pow(test_in[0], 2), std::pow(test_in[1], 3));

// A < B
X[0][0][0] = 1.0;
X[1][0][0] = 1.9;

// A > B
X[0][1][0] = 3.0;
X[1][1][0] = 2.0;

// A == B
X[0][2][0] = 1.0;
X[1][2][0] = 1.0;

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

X[0][1][1] = -2.0;
X[1][1][1] = -3.0;

// A < B
X[0][2][1] = 1.0;
X[1][2][1] = 2.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));

// A < B
BOOST_TEST(Y[0][0][0] == 3 * std::pow(test_in[1], 2) * X[1][0][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][0][1] ==
1.0 / 2.0 * 6.0 * test_in[1] * X[1][0][0] * X[1][0][0] +
3.0 * std::pow(test_in[1], 2.0) * X[1][0][1],
tt::tolerance(tol));

// A < B
BOOST_TEST(Y[0][1][0] == 2 * test_in[0] * X[0][1][0], tt::tolerance(tol));
BOOST_TEST(Y[0][1][1] == 1.0 / 2.0 * 2.0 * X[0][1][0] * X[0][1][0] +
2.0 * test_in[0] * X[0][1][1],
tt::tolerance(tol));

// A == B, A < B
BOOST_TEST(Y[0][2][0] == 3 * std::pow(test_in[1], 2) * X[1][2][0],
tt::tolerance(tol));
BOOST_TEST(Y[0][2][1] ==
1.0 / 2.0 * 6.0 * test_in[1] * X[1][2][0] * X[1][2][0] +
3.0 * std::pow(test_in[1], 2.0) * X[1][2][1],
tt::tolerance(tol));

myfree3(X);
myfree3(Y);
}
BOOST_AUTO_TEST_SUITE_END()

0 comments on commit cab7822

Please sign in to comment.