max/tug
1
0
Fork 0
mirror of https://git.gfz-potsdam.de/naaice/tug.git synced 2025-12-15 18:38:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Adjustment to set the threads via command line

Browse Source
This commit is contained in:
Hannes Signer 2023-08-23 18:35:26 +02:00
parent 6b8368d9f7
commit 1dbee6d8d9
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
examples/profiling_openmp.cpp
View File

@ -1,15 +1,17 @@
#include <tug/Simulation.hpp> #include <tug/Simulation.hpp>
#include <iostream> #include <iostream>
#include <fstream> #include <fstream>
#include <chrono>
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
int n[4] = {10, 20, 50, 100}; int n[4] = {100, 500, 1000, 2000};
int iterations[1] = {100}; int threads[10] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
int repetition = 10; int iterations[1] = {5};
int repetition = 1;
ofstream myfile; ofstream myfile;
myfile.open("time_measure_experiment_openmp_thread_1_EigenLU.csv"); myfile.open("testLarge.csv");
for (int i = 0; i < size(n); i++){ for (int i = 0; i < size(n); i++){
cout << "Grid size: " << n[i] << " x " << n[i] << endl << endl; cout << "Grid size: " << n[i] << " x " << n[i] << endl << endl;
@ -20,18 +22,25 @@ int main(int argc, char *argv[]) {
for (int k = 0; k < repetition; k++){ for (int k = 0; k < repetition; k++){
cout << "Wiederholung: " << k << endl; cout << "Wiederholung: " << k << endl;
Grid grid = Grid(n[i], n[i]); Grid grid = Grid(n[i], n[i]);
grid.setDomain(n[i], n[i]); grid.setDomain(1, 1);
MatrixXd concentrations = MatrixXd::Constant(n[i], n[i], 0); MatrixXd concentrations = MatrixXd::Constant(n[i], n[i], 0);
concentrations(5,5) = 1; concentrations(n[i]/2,n[i]/2) = 1;
grid.setConcentrations(concentrations); grid.setConcentrations(concentrations);
MatrixXd alpha = MatrixXd::Constant(n[i], n[i], 0.5); MatrixXd alpha = MatrixXd::Constant(n[i], n[i], 0.5);
Boundary bc = Boundary(grid); Boundary bc = Boundary(grid);
Simulation sim = Simulation(grid, bc, BTCS_APPROACH); Simulation sim = Simulation(grid, bc, BTCS_APPROACH);
sim.setSolver(EIGEN_LU_SOLVER); sim.setSolver(THOMAS_ALGORITHM_SOLVER);
if(argc == 2){
int numThreads = atoi(argv[1]);
sim.setNumberThreads(numThreads);
}
else{
sim.setNumberThreads(1);
}
sim.setTimestep(0.001); sim.setTimestep(0.001);
sim.setIterations(iterations[j]); sim.setIterations(iterations[j]);