795 lines
29 KiB
C++
795 lines
29 KiB
C++
/* ----------------------------------------------------------------------
|
|
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
|
|
https://www.lammps.org/, Sandia National Laboratories
|
|
LAMMPS Development team: developers@lammps.org
|
|
|
|
Copyright (2003) Sandia Corporation. Under the terms of Contract
|
|
DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
|
|
certain rights in this software. This software is distributed under
|
|
the GNU General Public License.
|
|
|
|
See the README file in the top-level LAMMPS directory.
|
|
------------------------------------------------------------------------- */
|
|
|
|
// unit tests for dihedral styles intended for molecular systems
|
|
|
|
#include "error_stats.h"
|
|
#include "test_config.h"
|
|
#include "test_config_reader.h"
|
|
#include "test_main.h"
|
|
#include "yaml_reader.h"
|
|
#include "yaml_writer.h"
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "atom.h"
|
|
#include "compute.h"
|
|
#include "dihedral.h"
|
|
#include "exceptions.h"
|
|
#include "fix.h"
|
|
#include "fmt/format.h"
|
|
#include "force.h"
|
|
#include "info.h"
|
|
#include "input.h"
|
|
#include "lammps.h"
|
|
#include "modify.h"
|
|
#include "platform.h"
|
|
#include "universe.h"
|
|
|
|
#include <cctype>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <mpi.h>
|
|
|
|
#include <map>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
using ::testing::HasSubstr;
|
|
using ::testing::StartsWith;
|
|
|
|
using namespace LAMMPS_NS;
|
|
|
|
void cleanup_lammps(LAMMPS *&lmp, const TestConfig &cfg)
|
|
{
|
|
platform::unlink(cfg.basename + ".restart");
|
|
platform::unlink(cfg.basename + ".data");
|
|
platform::unlink(cfg.basename + "-coeffs.in");
|
|
delete lmp;
|
|
lmp = nullptr;
|
|
}
|
|
|
|
LAMMPS *init_lammps(LAMMPS::argv &args, const TestConfig &cfg, const bool newton)
|
|
{
|
|
auto *lmp = new LAMMPS(args, MPI_COMM_WORLD);
|
|
|
|
// check if prerequisite styles are available
|
|
Info *info = new Info(lmp);
|
|
int nfail = 0;
|
|
for (const auto &prerequisite : cfg.prerequisites) {
|
|
std::string style = prerequisite.second;
|
|
|
|
// this is a test for dihedral styles, so if the suffixed
|
|
// version is not available, there is no reason to test.
|
|
if (prerequisite.first == "dihedral") {
|
|
if (lmp->suffix_enable) {
|
|
style += "/";
|
|
style += lmp->suffix;
|
|
}
|
|
}
|
|
|
|
if (!info->has_style(prerequisite.first, style)) ++nfail;
|
|
}
|
|
delete info;
|
|
if (nfail > 0) {
|
|
cleanup_lammps(lmp, cfg);
|
|
return nullptr;
|
|
}
|
|
|
|
// utility lambdas to improve readability
|
|
auto command = [&](const std::string &line) {
|
|
try {
|
|
lmp->input->one(line);
|
|
} catch (LAMMPSAbortException &ae) {
|
|
fprintf(stderr, "LAMMPS Error: %s\n", ae.what());
|
|
exit(2);
|
|
} catch (LAMMPSException &e) {
|
|
fprintf(stderr, "LAMMPS Error: %s\n", e.what());
|
|
exit(3);
|
|
} catch (fmt::format_error &fe) {
|
|
fprintf(stderr, "fmt::format_error: %s\n", fe.what());
|
|
exit(4);
|
|
} catch (std::exception &e) {
|
|
fprintf(stderr, "General exception: %s\n", e.what());
|
|
exit(5);
|
|
}
|
|
};
|
|
auto parse_input_script = [&](const std::string &filename) {
|
|
lmp->input->file(filename.c_str());
|
|
};
|
|
|
|
if (newton) {
|
|
command("variable newton_bond index on");
|
|
} else {
|
|
command("variable newton_bond index off");
|
|
}
|
|
|
|
command("variable input_dir index " + INPUT_FOLDER);
|
|
|
|
for (const auto &pre_command : cfg.pre_commands) {
|
|
command(pre_command);
|
|
}
|
|
|
|
std::string input_file = platform::path_join(INPUT_FOLDER, cfg.input_file);
|
|
parse_input_script(input_file);
|
|
|
|
command("dihedral_style " + cfg.dihedral_style);
|
|
|
|
for (const auto &dihedral_coeff : cfg.dihedral_coeff) {
|
|
command("dihedral_coeff " + dihedral_coeff);
|
|
}
|
|
|
|
for (const auto &post_command : cfg.post_commands) {
|
|
command(post_command);
|
|
}
|
|
|
|
command("run 0 post no");
|
|
command("variable write_data_pair index ii");
|
|
command("write_restart " + cfg.basename + ".restart");
|
|
command("write_data " + cfg.basename + ".data pair ${write_data_pair}");
|
|
command("write_coeff " + cfg.basename + "-coeffs.in");
|
|
|
|
return lmp;
|
|
}
|
|
|
|
void run_lammps(LAMMPS *lmp)
|
|
{
|
|
// utility lambda to improve readability
|
|
auto command = [&](const std::string &line) {
|
|
lmp->input->one(line);
|
|
};
|
|
|
|
command("fix 1 all nve");
|
|
command("compute pe all pe/atom dihedral");
|
|
command("compute sum all reduce sum c_pe");
|
|
command("thermo_style custom step temp pe press c_sum");
|
|
command("thermo 2");
|
|
command("run 4 post no");
|
|
}
|
|
|
|
void restart_lammps(LAMMPS *lmp, const TestConfig &cfg)
|
|
{
|
|
// utility lambda to improve readability
|
|
auto command = [&](const std::string &line) {
|
|
lmp->input->one(line);
|
|
};
|
|
|
|
command("clear");
|
|
command("read_restart " + cfg.basename + ".restart");
|
|
|
|
if (!lmp->force->dihedral) {
|
|
command("dihedral_style " + cfg.dihedral_style);
|
|
}
|
|
|
|
if ((cfg.dihedral_style.substr(0, 6) == "hybrid") || !lmp->force->dihedral->writedata) {
|
|
for (const auto &dihedral_coeff : cfg.dihedral_coeff) {
|
|
command("dihedral_coeff " + dihedral_coeff);
|
|
}
|
|
}
|
|
|
|
for (const auto &post_command : cfg.post_commands) {
|
|
command(post_command);
|
|
}
|
|
|
|
command("run 0 post no");
|
|
}
|
|
|
|
void data_lammps(LAMMPS *lmp, const TestConfig &cfg)
|
|
{
|
|
// utility lambdas to improve readability
|
|
auto command = [&](const std::string &line) {
|
|
lmp->input->one(line);
|
|
};
|
|
auto parse_input_script = [&](const std::string &filename) {
|
|
lmp->input->file(filename.c_str());
|
|
};
|
|
|
|
command("clear");
|
|
command("variable dihedral_style delete");
|
|
command("variable data_file delete");
|
|
command("variable newton_bond delete");
|
|
command("variable newton_bond index on");
|
|
|
|
for (const auto &pre_command : cfg.pre_commands) {
|
|
command(pre_command);
|
|
}
|
|
|
|
command("variable dihedral_style index '" + cfg.dihedral_style + "'");
|
|
command("variable data_file index " + cfg.basename + ".data");
|
|
|
|
// special treatment for dihedral styles charmm and charmmfsw
|
|
if (cfg.dihedral_style == "charmm") {
|
|
command("variable pair_style delete");
|
|
command("variable pair_style index 'lj/charmm/coul/charmm 7.0 8.0'");
|
|
} else if (cfg.dihedral_style == "charmmfsw") {
|
|
command("variable pair_style delete");
|
|
command("variable pair_style index 'lj/charmmfsw/coul/charmmfsh 7.0 8.0'");
|
|
}
|
|
|
|
std::string input_file = platform::path_join(INPUT_FOLDER, cfg.input_file);
|
|
parse_input_script(input_file);
|
|
|
|
for (const auto &dihedral_coeff : cfg.dihedral_coeff) {
|
|
command("dihedral_coeff " + dihedral_coeff);
|
|
}
|
|
for (const auto &post_command : cfg.post_commands) {
|
|
command(post_command);
|
|
}
|
|
command("run 0 post no");
|
|
}
|
|
|
|
// re-generate yaml file with current settings.
|
|
|
|
void generate_yaml_file(const char *outfile, const TestConfig &config)
|
|
{
|
|
// initialize system geometry
|
|
LAMMPS::argv args = {"DihedralStyle", "-log", "none", "-echo", "screen", "-nocite"};
|
|
|
|
LAMMPS *lmp = nullptr;
|
|
try {
|
|
lmp = init_lammps(args, config, true);
|
|
} catch (std::exception &e) {
|
|
FAIL() << e.what();
|
|
}
|
|
|
|
if (!lmp) {
|
|
std::cerr << "One or more prerequisite styles are not available "
|
|
"in this LAMMPS configuration:\n";
|
|
for (const auto &prerequisite : config.prerequisites) {
|
|
std::cerr << prerequisite.first << "_style " << prerequisite.second << "\n";
|
|
}
|
|
return;
|
|
}
|
|
|
|
const int natoms = lmp->atom->natoms;
|
|
std::string block;
|
|
|
|
YamlWriter writer(outfile);
|
|
|
|
// write yaml header
|
|
write_yaml_header(&writer, &test_config, lmp->version);
|
|
|
|
// dihedral_style
|
|
writer.emit("dihedral_style", config.dihedral_style);
|
|
|
|
// dihedral_coeff
|
|
block.clear();
|
|
for (const auto &dihedral_coeff : config.dihedral_coeff) {
|
|
block += dihedral_coeff + "\n";
|
|
}
|
|
writer.emit_block("dihedral_coeff", block);
|
|
|
|
// extract
|
|
block.clear();
|
|
for (auto data : config.extract)
|
|
block += fmt::format("{} {}\n", data.first, data.second);
|
|
writer.emit_block("extract", block);
|
|
|
|
// natoms
|
|
writer.emit("natoms", natoms);
|
|
|
|
// init_energy
|
|
writer.emit("init_energy", lmp->force->dihedral->energy);
|
|
|
|
// init_stress
|
|
auto *stress = lmp->force->dihedral->virial;
|
|
block = fmt::format("{:23.16e} {:23.16e} {:23.16e} {:23.16e} {:23.16e} {:23.16e}", stress[0],
|
|
stress[1], stress[2], stress[3], stress[4], stress[5]);
|
|
writer.emit_block("init_stress", block);
|
|
|
|
// init_forces
|
|
block.clear();
|
|
auto *f = lmp->atom->f;
|
|
for (int i = 1; i <= natoms; ++i) {
|
|
const int j = lmp->atom->map(i);
|
|
block += fmt::format("{:3} {:23.16e} {:23.16e} {:23.16e}\n", i, f[j][0], f[j][1], f[j][2]);
|
|
}
|
|
writer.emit_block("init_forces", block);
|
|
|
|
// do a few steps of MD
|
|
run_lammps(lmp);
|
|
|
|
// run_energy
|
|
writer.emit("run_energy", lmp->force->dihedral->energy);
|
|
|
|
// run_stress
|
|
stress = lmp->force->dihedral->virial;
|
|
block = fmt::format("{:23.16e} {:23.16e} {:23.16e} {:23.16e} {:23.16e} {:23.16e}", stress[0],
|
|
stress[1], stress[2], stress[3], stress[4], stress[5]);
|
|
writer.emit_block("run_stress", block);
|
|
|
|
block.clear();
|
|
f = lmp->atom->f;
|
|
for (int i = 1; i <= natoms; ++i) {
|
|
const int j = lmp->atom->map(i);
|
|
block += fmt::format("{:3} {:23.16e} {:23.16e} {:23.16e}\n", i, f[j][0], f[j][1], f[j][2]);
|
|
}
|
|
writer.emit_block("run_forces", block);
|
|
|
|
cleanup_lammps(lmp, config);
|
|
}
|
|
|
|
TEST(DihedralStyle, plain)
|
|
{
|
|
if (test_config.skip_tests.count(test_info_->name())) GTEST_SKIP();
|
|
|
|
LAMMPS::argv args = {"DihedralStyle", "-log", "none", "-echo", "screen", "-nocite"};
|
|
|
|
::testing::internal::CaptureStdout();
|
|
LAMMPS *lmp = nullptr;
|
|
try {
|
|
lmp = init_lammps(args, test_config, true);
|
|
} catch (std::exception &e) {
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
FAIL() << e.what();
|
|
}
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
|
|
if (!lmp) {
|
|
std::cerr << "One or more prerequisite styles are not available "
|
|
"in this LAMMPS configuration:\n";
|
|
for (auto &prerequisite : test_config.prerequisites) {
|
|
std::cerr << prerequisite.first << "_style " << prerequisite.second << "\n";
|
|
}
|
|
GTEST_SKIP();
|
|
}
|
|
|
|
EXPECT_THAT(output, StartsWith("LAMMPS ("));
|
|
EXPECT_THAT(output, HasSubstr("Loop time"));
|
|
|
|
// abort if running in parallel and not all atoms are local
|
|
const int nlocal = lmp->atom->nlocal;
|
|
ASSERT_EQ(lmp->atom->natoms, nlocal);
|
|
|
|
double epsilon = test_config.epsilon;
|
|
|
|
ErrorStats stats;
|
|
auto *dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("init_forces (newton on)", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("init_stress (newton on)", dihedral->virial, test_config.init_stress, epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "init_energy stats, newton on: " << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
run_lammps(lmp);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
EXPECT_FORCES("run_forces (newton on)", lmp->atom, test_config.run_forces, 10 * epsilon);
|
|
EXPECT_STRESS("run_stress (newton on)", dihedral->virial, test_config.run_stress, epsilon);
|
|
|
|
stats.reset();
|
|
auto *icompute = lmp->modify->get_compute_by_id("sum");
|
|
double energy = 0.0;
|
|
if (icompute) energy = icompute->compute_scalar();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.run_energy, epsilon);
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, energy, epsilon);
|
|
if (print_stats) std::cerr << "run_energy stats, newton on: " << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
try {
|
|
lmp = init_lammps(args, test_config, false);
|
|
} catch (std::exception &e) {
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
FAIL() << e.what();
|
|
}
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
// skip over these tests if newton bond is forced to be on
|
|
if (lmp->force->newton_bond == 0) {
|
|
dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("init_forces (newton off)", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("init_stress (newton off)", dihedral->virial, test_config.init_stress,
|
|
epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "init_energy stats, newton off:" << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
run_lammps(lmp);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
EXPECT_FORCES("run_forces (newton off)", lmp->atom, test_config.run_forces, 10 * epsilon);
|
|
EXPECT_STRESS("run_stress (newton off)", dihedral->virial, test_config.run_stress, epsilon);
|
|
|
|
stats.reset();
|
|
icompute = lmp->modify->get_compute_by_id("sum");
|
|
if (icompute) energy = icompute->compute_scalar();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.run_energy, epsilon);
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, energy, epsilon);
|
|
if (print_stats) std::cerr << "run_energy stats, newton off:" << stats << std::endl;
|
|
}
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
restart_lammps(lmp, test_config);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("restart_forces", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("restart_stress", dihedral->virial, test_config.init_stress, epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "restart_energy stats:" << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
data_lammps(lmp, test_config);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("data_forces", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("data_stress", dihedral->virial, test_config.init_stress, epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "data_energy stats:" << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
};
|
|
|
|
TEST(DihedralStyle, omp)
|
|
{
|
|
if (!Info::has_package("OPENMP")) GTEST_SKIP();
|
|
if (test_config.skip_tests.count(test_info_->name())) GTEST_SKIP();
|
|
|
|
LAMMPS::argv args = {"DihedralStyle", "-log", "none", "-echo", "screen", "-nocite",
|
|
"-pk", "omp", "4", "-sf", "omp"};
|
|
|
|
::testing::internal::CaptureStdout();
|
|
LAMMPS *lmp = nullptr;
|
|
try {
|
|
lmp = init_lammps(args, test_config, true);
|
|
} catch (std::exception &e) {
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
FAIL() << e.what();
|
|
}
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
|
|
if (!lmp) {
|
|
std::cerr << "One or more prerequisite styles with /omp suffix\n"
|
|
"are not available in this LAMMPS configuration:\n";
|
|
for (auto &prerequisite : test_config.prerequisites) {
|
|
std::cerr << prerequisite.first << "_style " << prerequisite.second << "\n";
|
|
}
|
|
GTEST_SKIP();
|
|
}
|
|
|
|
EXPECT_THAT(output, StartsWith("LAMMPS ("));
|
|
EXPECT_THAT(output, HasSubstr("Loop time"));
|
|
|
|
// abort if running in parallel and not all atoms are local
|
|
const int nlocal = lmp->atom->nlocal;
|
|
ASSERT_EQ(lmp->atom->natoms, nlocal);
|
|
|
|
// relax error a bit for OPENMP package
|
|
double epsilon = 5.0 * test_config.epsilon;
|
|
|
|
ErrorStats stats;
|
|
auto *dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("init_forces (newton on)", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("init_stress (newton on)", dihedral->virial, test_config.init_stress,
|
|
10 * epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "init_energy stats, newton on: " << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
run_lammps(lmp);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
EXPECT_FORCES("run_forces (newton on)", lmp->atom, test_config.run_forces, 10 * epsilon);
|
|
EXPECT_STRESS("run_stress (newton on)", dihedral->virial, test_config.run_stress, 10 * epsilon);
|
|
|
|
stats.reset();
|
|
auto *icompute = lmp->modify->get_compute_by_id("sum");
|
|
double energy = 0.0;
|
|
if (icompute) energy = icompute->compute_scalar();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.run_energy, epsilon);
|
|
// TODO: this is currently broken for OPENMP with dihedral style hybrid
|
|
// needs to be fixed in the main code somewhere. Not sure where, though.
|
|
if (test_config.dihedral_style.substr(0, 6) != "hybrid")
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, energy, epsilon);
|
|
if (print_stats) std::cerr << "run_energy stats, newton on: " << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
try {
|
|
lmp = init_lammps(args, test_config, false);
|
|
} catch (std::exception &e) {
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
FAIL() << e.what();
|
|
}
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
// skip over these tests if newton bond is forced to be on
|
|
if (lmp->force->newton_bond == 0) {
|
|
dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("init_forces (newton off)", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("init_stress (newton off)", dihedral->virial, test_config.init_stress,
|
|
10 * epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "init_energy stats, newton off:" << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
run_lammps(lmp);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
EXPECT_FORCES("run_forces (newton off)", lmp->atom, test_config.run_forces, 10 * epsilon);
|
|
EXPECT_STRESS("run_stress (newton off)", dihedral->virial, test_config.run_stress,
|
|
10 * epsilon);
|
|
|
|
stats.reset();
|
|
icompute = lmp->modify->get_compute_by_id("sum");
|
|
if (icompute) energy = icompute->compute_scalar();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.run_energy, epsilon);
|
|
// TODO: this is currently broken for OPENMP with dihedral style hybrid
|
|
// needs to be fixed in the main code somewhere. Not sure where, though.
|
|
if (test_config.dihedral_style.substr(0, 6) != "hybrid")
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, energy, epsilon);
|
|
if (print_stats) std::cerr << "run_energy stats, newton off:" << stats << std::endl;
|
|
}
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
};
|
|
|
|
TEST(DihedralStyle, kokkos_omp)
|
|
{
|
|
if (!Info::has_package("KOKKOS")) GTEST_SKIP();
|
|
if (test_config.skip_tests.count(test_info_->name())) GTEST_SKIP();
|
|
// test either OpenMP or Serial
|
|
if (!Info::has_accelerator_feature("KOKKOS", "api", "serial") &&
|
|
!Info::has_accelerator_feature("KOKKOS", "api", "openmp"))
|
|
GTEST_SKIP();
|
|
// if KOKKOS has GPU support enabled, it *must* be used. We cannot test OpenMP only.
|
|
if (Info::has_accelerator_feature("KOKKOS", "api", "cuda") ||
|
|
Info::has_accelerator_feature("KOKKOS", "api", "hip") ||
|
|
Info::has_accelerator_feature("KOKKOS", "api", "sycl"))
|
|
GTEST_SKIP() << "Cannot test KOKKOS/OpenMP with GPU support enabled";
|
|
|
|
LAMMPS::argv args = {"DihedralStyle", "-log", "none", "-echo", "screen",
|
|
"-nocite", "-k", "on", "t", "4",
|
|
"-sf", "kk"};
|
|
// fall back to serial if openmp is not available
|
|
if (!Info::has_accelerator_feature("KOKKOS", "api", "openmp")) args[9] = "1";
|
|
|
|
::testing::internal::CaptureStdout();
|
|
LAMMPS *lmp = nullptr;
|
|
try {
|
|
lmp = init_lammps(args, test_config, true);
|
|
} catch (std::exception &e) {
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
FAIL() << e.what();
|
|
}
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
|
|
if (!lmp) {
|
|
std::cerr << "One or more prerequisite styles with /kk suffix\n"
|
|
"are not available in this LAMMPS configuration:\n";
|
|
for (auto &prerequisite : test_config.prerequisites) {
|
|
std::cerr << prerequisite.first << "_style " << prerequisite.second << "\n";
|
|
}
|
|
GTEST_SKIP();
|
|
}
|
|
|
|
EXPECT_THAT(output, StartsWith("LAMMPS ("));
|
|
EXPECT_THAT(output, HasSubstr("Loop time"));
|
|
|
|
// abort if running in parallel and not all atoms are local
|
|
const int nlocal = lmp->atom->nlocal;
|
|
ASSERT_EQ(lmp->atom->natoms, nlocal);
|
|
|
|
// relax error a bit for KOKKOS package
|
|
double epsilon = 5.0 * test_config.epsilon;
|
|
|
|
ErrorStats stats;
|
|
auto dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("init_forces (newton on)", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("init_stress (newton on)", dihedral->virial, test_config.init_stress,
|
|
10 * epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "init_energy stats, newton on: " << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
run_lammps(lmp);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
EXPECT_FORCES("run_forces (newton on)", lmp->atom, test_config.run_forces, 10 * epsilon);
|
|
EXPECT_STRESS("run_stress (newton on)", dihedral->virial, test_config.run_stress, 10 * epsilon);
|
|
|
|
stats.reset();
|
|
auto *icompute = lmp->modify->get_compute_by_id("sum");
|
|
if (icompute) icompute->compute_scalar();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.run_energy, epsilon);
|
|
|
|
// FIXME: this is currently broken ??? for KOKKOS with dihedral style hybrid
|
|
// needs to be fixed in the main code somewhere. Not sure where, though.
|
|
// if (test_config.dihedral_style.substr(0, 6) != "hybrid")
|
|
// EXPECT_FP_LE_WITH_EPS(dihedral->energy, energy, epsilon);
|
|
// if (print_stats) std::cerr << "run_energy stats, newton on: " << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
try {
|
|
lmp = init_lammps(args, test_config, false);
|
|
} catch (std::exception &e) {
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
FAIL() << e.what();
|
|
}
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
// skip over these tests if newton bond is forced to be on
|
|
if (lmp->force->newton_bond == 0) {
|
|
dihedral = lmp->force->dihedral;
|
|
|
|
EXPECT_FORCES("init_forces (newton off)", lmp->atom, test_config.init_forces, epsilon);
|
|
EXPECT_STRESS("init_stress (newton off)", dihedral->virial, test_config.init_stress,
|
|
10 * epsilon);
|
|
|
|
stats.reset();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.init_energy, epsilon);
|
|
if (print_stats) std::cerr << "init_energy stats, newton off:" << stats << std::endl;
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
run_lammps(lmp);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
EXPECT_FORCES("run_forces (newton off)", lmp->atom, test_config.run_forces, 10 * epsilon);
|
|
EXPECT_STRESS("run_stress (newton off)", dihedral->virial, test_config.run_stress,
|
|
10 * epsilon);
|
|
|
|
stats.reset();
|
|
auto *icompute = lmp->modify->get_compute_by_id("sum");
|
|
if (icompute) icompute->compute_scalar();
|
|
EXPECT_FP_LE_WITH_EPS(dihedral->energy, test_config.run_energy, epsilon);
|
|
|
|
// FIXME: this is currently broken ??? for KOKKOS with dihedral style hybrid
|
|
// needs to be fixed in the main code somewhere. Not sure where, though.
|
|
// if (test_config.dihedral_style.substr(0, 6) != "hybrid")
|
|
// EXPECT_FP_LE_WITH_EPS(dihedral->energy, energy, epsilon);
|
|
|
|
if (print_stats) std::cerr << "run_energy stats, newton off:" << stats << std::endl;
|
|
}
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
};
|
|
|
|
TEST(DihedralStyle, numdiff)
|
|
{
|
|
if (!Info::has_package("EXTRA-FIX")) GTEST_SKIP();
|
|
if (test_config.skip_tests.count(test_info_->name())) GTEST_SKIP();
|
|
|
|
LAMMPS::argv args = {"DihedralStyle", "-log", "none", "-echo", "screen", "-nocite"};
|
|
|
|
::testing::internal::CaptureStdout();
|
|
LAMMPS *lmp = nullptr;
|
|
try {
|
|
lmp = init_lammps(args, test_config, true);
|
|
} catch (std::exception &e) {
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
FAIL() << e.what();
|
|
}
|
|
std::string output = ::testing::internal::GetCapturedStdout();
|
|
if (verbose) std::cout << output;
|
|
|
|
if (!lmp) {
|
|
std::cerr << "One or more prerequisite styles are not available "
|
|
"in this LAMMPS configuration:\n";
|
|
for (auto &prerequisite : test_config.prerequisites) {
|
|
std::cerr << prerequisite.first << "_style " << prerequisite.second << "\n";
|
|
}
|
|
GTEST_SKIP();
|
|
}
|
|
|
|
EXPECT_THAT(output, StartsWith("LAMMPS ("));
|
|
EXPECT_THAT(output, HasSubstr("Loop time"));
|
|
|
|
// abort if running in parallel and not all atoms are local
|
|
const int nlocal = lmp->atom->nlocal;
|
|
ASSERT_EQ(lmp->atom->natoms, nlocal);
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
lmp->input->one("fix diff all numdiff 2 6.05504e-6");
|
|
lmp->input->one("run 2 post no");
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
Fix *ifix = lmp->modify->get_fix_by_id("diff");
|
|
if (ifix) {
|
|
double epsilon = test_config.epsilon * 5.0e8;
|
|
ErrorStats stats;
|
|
double **f1 = lmp->atom->f;
|
|
double **f2 = ifix->array_atom;
|
|
SCOPED_TRACE("EXPECT FORCES: numdiff");
|
|
for (int i = 0; i < nlocal; ++i) {
|
|
EXPECT_FP_LE_WITH_EPS(f1[i][0], f2[i][0], epsilon);
|
|
EXPECT_FP_LE_WITH_EPS(f1[i][1], f2[i][1], epsilon);
|
|
EXPECT_FP_LE_WITH_EPS(f1[i][2], f2[i][2], epsilon);
|
|
}
|
|
if (print_stats)
|
|
std::cerr << "numdiff stats: " << stats << " epsilon: " << epsilon << std::endl;
|
|
}
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
}
|
|
|
|
TEST(DihedralStyle, extract)
|
|
{
|
|
if (test_config.skip_tests.count(test_info_->name())) GTEST_SKIP();
|
|
|
|
LAMMPS::argv args = {"DihedralStyle", "-log", "none", "-echo", "screen", "-nocite"};
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
LAMMPS *lmp = nullptr;
|
|
try {
|
|
lmp = init_lammps(args, test_config, true);
|
|
} catch (std::exception &e) {
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
FAIL() << e.what();
|
|
}
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
|
|
if (!lmp) {
|
|
std::cerr << "One or more prerequisite styles are not available "
|
|
"in this LAMMPS configuration:\n";
|
|
for (auto prerequisite : test_config.prerequisites) {
|
|
std::cerr << prerequisite.first << "_style " << prerequisite.second << "\n";
|
|
}
|
|
GTEST_SKIP();
|
|
}
|
|
|
|
auto *dihedral = lmp->force->dihedral;
|
|
void *ptr = nullptr;
|
|
int dim = 0;
|
|
for (auto extract : test_config.extract) {
|
|
ptr = dihedral->extract(extract.first.c_str(), dim);
|
|
EXPECT_NE(ptr, nullptr);
|
|
EXPECT_EQ(dim, extract.second);
|
|
}
|
|
ptr = dihedral->extract("does_not_exist", dim);
|
|
EXPECT_EQ(ptr, nullptr);
|
|
|
|
if (!verbose) ::testing::internal::CaptureStdout();
|
|
cleanup_lammps(lmp, test_config);
|
|
if (!verbose) ::testing::internal::GetCapturedStdout();
|
|
}
|