Fixing cuda 10.1 build errors.

UO-OACISS · Mar 22, 2021 · 81a8d90 · 81a8d90
1 parent abe99f9
commit 81a8d90
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/apex/activity_trace_async.cpp b/src/apex/activity_trace_async.cpp
@@ -1002,10 +1002,10 @@ bool getBytesIfMalloc(CUpti_CallbackId id, const void* params, std::string conte
     bool onHost = false;
     bool managed = false;
     void* ptr = nullptr;
-    static std::atomic<size_t> totalAllocated = 0.0;
+    static std::atomic<size_t> totalAllocated{0};
     static std::unordered_map<void*,size_t> memoryMap;
     std::mutex mapMutex;
-    static std::atomic<size_t> hostTotalAllocated = 0.0;
+    static std::atomic<size_t> hostTotalAllocated{0};
     static std::unordered_map<void*,size_t> hostMemoryMap;
     std::mutex hostMapMutex;
     bool free = false;
@@ -1087,11 +1087,13 @@ bool getBytesIfMalloc(CUpti_CallbackId id, const void* params, std::string conte
                 onHost = true;
                 break;
             }
+#ifdef CUPTI_DRIVER_TRACE_CBID_cuMemAddressFree
             case CUPTI_DRIVER_TRACE_CBID_cuMemAddressFree: {
                 ptr = (void*)((cuMemAddressFree_params_st*)(params))->ptr;
                 free = true;
                 break;
             }
+#endif
             case CUPTI_DRIVER_TRACE_CBID_cuMemFree: {
                 size_t tmp = (size_t)((cuMemFree_params_st*)(params))->dptr;
                 ptr = (void*)(tmp);