mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 17:17:14 +08:00
92 lines
3.4 KiB
C++
92 lines
3.4 KiB
C++
#include <bench/spbench/spbenchsolver.h>
|
|
|
|
void bench_printhelp() {
|
|
cout << " \nbenchsolver : performs a benchmark of all the solvers available "
|
|
"in Eigen \n\n";
|
|
cout << " MATRIX FOLDER : \n";
|
|
cout << " The matrices for the benchmark should be collected in a folder "
|
|
"specified with an environment variable EIGEN_MATRIXDIR \n";
|
|
cout << " The matrices are stored using the matrix market coordinate format "
|
|
"\n";
|
|
cout << " The matrix and associated right-hand side (rhs) files are named "
|
|
"respectively \n";
|
|
cout << " as MatrixName.mtx and MatrixName_b.mtx. If the rhs does not exist, "
|
|
"a random one is generated. \n";
|
|
cout << " If a matrix is SPD, the matrix should be named as "
|
|
"MatrixName_SPD.mtx \n";
|
|
cout << " If a true solution exists, it should be named as MatrixName_x.mtx; "
|
|
"\n";
|
|
cout << " it will be used to compute the norm of the error relative to the "
|
|
"computed solutions\n\n";
|
|
cout << " OPTIONS : \n";
|
|
cout << " -h or --help \n print this help and return\n\n";
|
|
cout << " -d matrixdir \n Use matrixdir as the matrix folder instead of "
|
|
"the one specified in the environment variable EIGEN_MATRIXDIR\n\n";
|
|
cout << " -o outputfile.xml \n Output the statistics to a xml file \n\n";
|
|
cout << " --eps <RelErr> Sets the relative tolerance for iterative solvers "
|
|
"(default 1e-08) \n\n";
|
|
cout << " --maxits <MaxIts> Sets the maximum number of iterations (default "
|
|
"1000) \n\n";
|
|
}
|
|
int main(int argc, char** args) {
|
|
bool help =
|
|
(get_options(argc, args, "-h") || get_options(argc, args, "--help"));
|
|
if (help) {
|
|
bench_printhelp();
|
|
return 0;
|
|
}
|
|
|
|
// Get the location of the test matrices
|
|
string matrix_dir;
|
|
if (!get_options(argc, args, "-d", &matrix_dir)) {
|
|
if (getenv("EIGEN_MATRIXDIR") == NULL) {
|
|
std::cerr << "Please, specify the location of the matrices with -d "
|
|
"mat_folder or the environment variable EIGEN_MATRIXDIR \n";
|
|
std::cerr
|
|
<< " Run with --help to see the list of all the available options \n";
|
|
return -1;
|
|
}
|
|
matrix_dir = getenv("EIGEN_MATRIXDIR");
|
|
}
|
|
|
|
std::ofstream statbuf;
|
|
string statFile;
|
|
|
|
// Get the file to write the statistics
|
|
bool statFileExists = get_options(argc, args, "-o", &statFile);
|
|
if (statFileExists) {
|
|
statbuf.open(statFile.c_str(), std::ios::out);
|
|
if (statbuf.good()) {
|
|
statFileExists = true;
|
|
printStatheader(statbuf);
|
|
statbuf.close();
|
|
} else
|
|
std::cerr << "Unable to open the provided file for writing... \n";
|
|
}
|
|
|
|
// Get the maximum number of iterations and the tolerance
|
|
int maxiters = 1000;
|
|
double tol = 1e-08;
|
|
string inval;
|
|
if (get_options(argc, args, "--eps", &inval)) tol = atof(inval.c_str());
|
|
if (get_options(argc, args, "--maxits", &inval))
|
|
maxiters = atoi(inval.c_str());
|
|
|
|
string current_dir;
|
|
// Test the real-arithmetics matrices
|
|
Browse_Matrices<double>(matrix_dir, statFileExists, statFile, maxiters, tol);
|
|
|
|
// Test the complex-arithmetics matrices
|
|
Browse_Matrices<std::complex<double> >(matrix_dir, statFileExists, statFile,
|
|
maxiters, tol);
|
|
|
|
if (statFileExists) {
|
|
statbuf.open(statFile.c_str(), std::ios::app);
|
|
statbuf << "</BENCH> \n";
|
|
cout << "\n Output written in " << statFile << " ...\n";
|
|
statbuf.close();
|
|
}
|
|
|
|
return 0;
|
|
}
|