Fixing label for host-allocated memory in Cuda

UO-OACISS · Sep 1, 2020 · 55bacb3 · 55bacb3
1 parent 7c80b79
commit 55bacb3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/apex/activity_trace_async.cpp b/src/apex/activity_trace_async.cpp
@@ -672,6 +672,11 @@ bool getBytesIfMalloc(CUpti_CallbackId id, const void* params, std::string conte
         }
         case CUPTI_RUNTIME_TRACE_CBID_cudaMallocHost_v3020: {
             bytes = ((cudaMallocHost_v3020_params_st*)(params))->size;
+            // we have a special case - handle it differently...
+            double value = (double)(bytes);
+            store_counter_data("Host: Page-locked Bytes Allocated", context,
+                apex::profiler::get_time_ns(), value);
+            return true;
             break;
         }
         case CUPTI_RUNTIME_TRACE_CBID_cudaMalloc3D_v3020: {