1 files changed, 13 insertions, 2 deletions
diff --git a/runtime/contrib/heap_trace/src/trace.h b/runtime/contrib/heap_trace/src/trace.h
index f03a65a58..647c51d54 100644
--- a/runtime/contrib/heap_trace/src/trace.h
+++ b/runtime/contrib/heap_trace/src/trace.h
@@ -25,6 +25,17 @@
 
 class Trace
 {
+  struct MemoryTraits
+  {
+    size_t ref_counter;
+    size_t size;
+
+    MemoryTraits(size_t init_counter_value, size_t size_of_allocated_memory)
+        : ref_counter(init_counter_value), size(size_of_allocated_memory)
+    {
+    }
+  };
+
 public:
   class Guard
   {
@@ -37,7 +48,7 @@ public:
     void markTraceAsReady() { _is_trace_not_available = false; }
     void markTraceAsNotReady() { _is_trace_not_available = true; }
     void signalizeAboutPossibleRecursion() { _is_recursion_detected = true; }
-    void signalizeThatDangerOfRecursionHAsPassed() { _is_recursion_detected = false; }
+    void signalizeThatDangerOfRecursionHasPassed() { _is_recursion_detected = false; }
 
   private:
     static bool _is_trace_not_available;
@@ -69,7 +80,7 @@ private:
   size_t _total_deallocated_bytes_on_gpu = 0;
   size_t _peak_heap_usage_on_gpu = 0;
   std::unordered_map<void *, size_t> _memory_in_use_on_cpu;
-  std::unordered_map<cl_mem, size_t> _memory_in_use_on_gpu;
+  std::unordered_map<cl_mem, MemoryTraits> _memory_in_use_on_gpu;
 };
 
 #endif // !TRACE_H