Call cudaProfilerStop() when destroying a context so profiling will work correctly

11b86e0d · Peter Eastman · a106d7d1 · 11b86e0d
Commit 11b86e0d authored Apr 12, 2013 by Peter Eastman
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

platforms/cuda/src/CudaContext.cpp platforms/cuda/src/CudaContext.cpp +4 -1

No files found.
--- a/platforms/cuda/src/CudaContext.cpp
+++ b/platforms/cuda/src/CudaContext.cpp
@@ -47,6 +47,7 @@
 #include <iostream>
 #include <sstream>
 #include <typeinfo>
+#include <cudaProfiler.h>
 #define CHECK_RESULT(result) CHECK_RESULT2(result, errorMessage);
@@ -253,8 +254,10 @@ CudaContext::~CudaContext() {
    if (thread != NULL)
        delete thread;
    string errorMessage = "Error deleting Context";
-    if (contextIsValid)
+    if (contextIsValid) {
+        cuProfilerStop();
        CHECK_RESULT(cuCtxDestroy(context));
+    }
    contextIsValid = false;
 }