From 2a099da5b758ab697e9508817f7282e6d7868838 Mon Sep 17 00:00:00 2001 From: Maximilian Gaul Date: Fri, 3 Dec 2021 08:13:43 +0100 Subject: [PATCH] Started cuda profiling, added first result to logbook --- src/force.cu | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/src/force.cu b/src/force.cu index 8724342..a2c397e 100644 --- a/src/force.cu +++ b/src/force.cu @@ -24,6 +24,7 @@ #include #include #include +#include #include #include @@ -122,6 +123,8 @@ double computeForce( MD_FLOAT epsilon = param->epsilon; #endif + cudaProfilerStart(); + for(int i = 0; i < Nlocal; i++) { fx[i] = 0.0; fy[i] = 0.0; @@ -129,7 +132,12 @@ double computeForce( } const char *num_threads_env = getenv("NUM_THREADS"); - const int num_threads = atoi(num_threads_env); + int num_threads = 0; + if(num_threads_env == nullptr) + num_threads = 2; + else { + num_threads = atoi(num_threads_env); + } Atom c_atom; c_atom.Natoms = atom->Natoms; @@ -225,6 +233,8 @@ double computeForce( cudaFree(c_neighs); cudaFree(c_neigh_numneigh); + cudaProfilerStop(); + LIKWID_MARKER_STOP("force"); double E = getTimeStamp();