diff --git a/CMakeLists.txt b/CMakeLists.txt index 7323399..9fe066c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -88,6 +88,7 @@ if (BUILD_SAMPLES) add_subdirectory(samples/ctest) add_subdirectory(samples/cpptest) add_subdirectory(samples/mpitest) + add_subdirectory(samples/benchmark) endif() if (BUILD_STANDALONE) diff --git a/samples/benchmark/CMakeLists.txt b/samples/benchmark/CMakeLists.txt new file mode 100644 index 0000000..3dc2b1e --- /dev/null +++ b/samples/benchmark/CMakeLists.txt @@ -0,0 +1,3 @@ +## benchmark +add_executable(benchmark main.cc) +target_link_libraries(benchmark astaroth_core astaroth_utils) diff --git a/samples/benchmark/main.cc b/samples/benchmark/main.cc new file mode 100644 index 0000000..0bd798f --- /dev/null +++ b/samples/benchmark/main.cc @@ -0,0 +1,122 @@ +/* + Copyright (C) 2014-2020, Johannes Pekkila, Miikka Vaisala. + + This file is part of Astaroth. + + Astaroth is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + Astaroth is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with Astaroth. If not, see . +*/ +/** + Running: benchmark -np +*/ +#include "astaroth.h" +#include "astaroth_utils.h" + +#include "timer_hires.h" + +#if AC_MPI_ENABLED + +#include + +#include +#include + +int +main(void) +{ + MPI_Init(NULL, NULL); + int nprocs, pid; + MPI_Comm_size(MPI_COMM_WORLD, &nprocs); + MPI_Comm_rank(MPI_COMM_WORLD, &pid); + + // CPU alloc + AcMeshInfo info; + acLoadConfig(AC_DEFAULT_CONFIG, &info); + + /* + AcMesh model, candidate; + if (pid == 0) { + acMeshCreate(info, &model); + acMeshCreate(info, &candidate); + acMeshRandomize(&model); + acMeshRandomize(&candidate); + }*/ + + // GPU alloc & compute + acGridInit(info); + /* + acGridLoadMesh(model, STREAM_DEFAULT); + + acGridIntegrate(STREAM_DEFAULT, FLT_EPSILON); + acGridPeriodicBoundconds(STREAM_DEFAULT); + + acGridStoreMesh(STREAM_DEFAULT, &candidate); + + // Verify + if (pid == 0) { + acModelIntegrateStep(model, FLT_EPSILON); + acMeshApplyPeriodicBounds(&model); + + AcResult retval = acVerifyMesh(model, candidate); + acMeshDestroy(&model); + acMeshDestroy(&candidate); + + if (retval != AC_SUCCESS) { + fprintf(stderr, "Failures found, benchmark invalid. Skipping\n"); + return EXIT_FAILURE; + } + }*/ + + // Warmup + for (size_t i = 0; i < 10; ++i) + acGridIntegrate(STREAM_DEFAULT, FLT_EPSILON); + + // Benchmark + Timer t; + const AcReal dt = FLT_EPSILON; + const size_t num_iters = 100; + const double nth_percentile = 0.90; + + std::vector results; // ms + results.reserve(num_iters); + + for (size_t i = 0; i < num_iters; ++i) { + acGridSynchronizeStream(STREAM_ALL); + timer_reset(&t); + acGridSynchronizeStream(STREAM_ALL); + acGridIntegrate(STREAM_DEFAULT, dt); + acGridSynchronizeStream(STREAM_ALL); + results.push_back(timer_diff_nsec(t) / 1e6); + } + + std::sort(results.begin(), results.end(), + [](const double& a, const double& b) { return a < b; }); + fprintf(stdout, + "Integration step time %g ms (%gth percentile)--------------------------------------\n", + results[nth_percentile * num_iters], 100 * nth_percentile); + acGridSynchronizeStream(STREAM_ALL); + + acGridQuit(); + MPI_Finalize(); + return EXIT_SUCCESS; +} + +#else +int +main(void) +{ + printf("The library was built without MPI support, cannot run mpitest. Rebuild Astaroth with " + "cmake -DMPI_ENABLED=ON .. to enable.\n"); + return EXIT_FAILURE; +} +#endif // AC_MPI_ENABLES