poisson_WS.cc
ca0d989eb263master

poisson_WS.cc
View Options

			/* -------------------------------------------------------------------------- */
			#include "simulation.hh"
			/* -------------------------------------------------------------------------- */
			#include <chrono>
			#include <iostream>
			#include <sstream>
			#include <tuple>
			#include <chrono>
			#include <cmath>
			/* -------------------------------------------------------------------------- */
			#include <mpi.h>
			/* -------------------------------------------------------------------------- */

			typedef std::chrono::high_resolution_clock clk;
			typedef std::chrono::duration<double> second;

			static void usage(const std::string & prog_name) {
			std::cerr << prog_name << " <grid_size> <n_iter>" << std::endl;
			exit(0);
			}

			int main(int argc, char * argv[]) {
			MPI_Init(&argc, &argv);
			int prank, psize;

			MPI_Comm_rank(MPI_COMM_WORLD, &prank);
			MPI_Comm_size(MPI_COMM_WORLD, &psize);

			if (argc != 3) usage(argv[0]);

			std::stringstream arg_0(argv[1]), arg_1(argv[2]);
			int n_iter;
			int N;
			arg_0 >> N;
			arg_1 >> n_iter;

			if(arg_0.fail() \|\| arg_1.fail()) usage(argv[0]);

			// Compute slice parameters
			int n_loc = (int)(4096/std::sqrt(psize));
			N = (int)(4096*std::sqrt(psize));

			Simulation simu(n_loc, N, n_iter, prank, psize);

			simu.set_initial_conditions();

			float l2;
			int k;

			auto start = clk::now();
			std::tie(l2, k) = simu.compute();
			auto end = clk::now();

			second time = end - start;

			if(prank == 0)
			std::cout << psize << " " << N << " "
			<< k << " " << std::scientific << l2 << " "
			<< time.count() << std::endl;

			MPI_Finalize();

			return 0;
			}