Add NVTX markers to the code

[hercules2020/kcf.git] / src / debug.h

diff --git a/src/debug.h b/src/debug.h
index 6ad1d18c1d70ffd17a7f42d12619e0fb2e93c5df..07c6ee77a532ebc5bf2208e476441b26792c138b 100644 (file)
--- a/src/debug.h
+++ b/src/debug.h
@@ -11,6 +11,7 @@
 
 #ifdef CUFFT
 #include <cufft.h>
+#include "nvToolsExt.h"
 #endif
 
 
@@ -52,6 +53,9 @@ class DbgTracer {
       public:
         FTrace(DbgTracer &dt, const char *fn, const char *format, ...) : t(dt), funcName(fn)
         {
+#ifdef CUFFT
+            nvtxRangePushA(fn);
+#endif
             if (!t.debug) return;
             char *arg;
             va_list vl;
@@ -65,6 +69,9 @@ class DbgTracer {
         }
         ~FTrace()
         {
+#ifdef CUFFT
+            nvtxRangePop();
+#endif
             if (!t.debug) return;
             t.indentLvl--;
             std::cerr << t.indent() << "}" << std::endl;