Add MPI_Wtime() in MPIWrapper

dftfeDevelopers · Jan 5, 2024 · 59734ab · 59734ab
1 parent 85231a6
commit 59734ab
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/utils/MPIWrapper.cpp b/src/utils/MPIWrapper.cpp
@@ -26,6 +26,7 @@
 #include <utils/Exceptions.h>
 #include <utils/MemoryTransfer.h>
 #include <complex>
+#include <ctime>
 namespace dftefe
 {
   namespace utils
@@ -219,6 +220,12 @@ namespace dftefe
         return ::MPI_Finalized(flag);
       }
 
+      double
+      MPIWtime(void)
+      {
+        return ::MPI_Wtime();
+      }
+
       bool
       MPIErrIsSuccess(int errCode)
       {
@@ -475,6 +482,12 @@ namespace dftefe
         return MPISuccess;
       }
 
+      double
+      MPIWtime(void)
+      {
+        return std::time(nullptr);
+      }
+
       bool
       MPIErrIsSuccess(int errCode)
       {

diff --git a/src/utils/MPIWrapper.h b/src/utils/MPIWrapper.h
@@ -255,6 +255,9 @@ namespace dftefe
       int
       MPIFinalize(void);
 
+      double
+      MPIWtime(void);
+
       int
       MPIFinalized(int *flag);
-Original file line number
+Diff line change
@@ Expand Up / @@ -255,6 +255,9 @@ namespace dftefe @@
           int
           MPIFinalize(void);
+          double
+          MPIWtime(void);
           int
           MPIFinalized(int *flag);
@@ Expand Down @@