Commit 8d000b81 authored by Giovanni La Mura's avatar Giovanni La Mura
Browse files

Use NVTX profiler in cfrfme.cpp

parent d320c140

src/trapping/cfrfme.cpp

+10 −0
Original line number Diff line number Diff line
@@ -56,6 +56,10 @@ 
#include "../include/tra_subs.h"

#endif



#ifdef USE_NVTX

#include <nvtx3/nvToolsExt.h>

#endif



using namespace std;



/*! \brief C++ implementation of FRFME
@@ -64,6 +68,9 @@ using namespace std; 
 *  \param output_path: `string` Directory to write the output files in.

 */

void frfme(string data_file, string output_path) {

#ifdef USE_NVTX

  nvtxRangePush("Running frfme()");

#endif

  string tfrfme_name = output_path + "/c_TFRFME.hd5";

  TFRFME *tfrfme = NULL;

  Swap1 *tt1 = NULL;
@@ -410,4 +417,7 @@ void frfme(string data_file, string output_path) { 
  if (wk != NULL) delete[] wk;

  if (tt1 != NULL) delete tt1;

  printf("FRFME: Done.\n");

#ifdef USE_NVTX

  nvtxRangePop();

#endif

}