Merge pull request #257 from vlkale/spc-tm-stk-alloc-null-ret

Check for de/allocations pointing to null
kokkos · Jun 10, 2024 · 3cee564 · 3cee564
2 parents 0a5d81d + a8c9ca0
commit 3cee564
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/profiling/space-time-stack/kp_space_time_stack.cpp b/profiling/space-time-stack/kp_space_time_stack.cpp
@@ -521,12 +521,20 @@ struct Allocations {
   Allocations() : total_size(0) {}
   void allocate(std::string&& name, const void* ptr, std::uint64_t size,
                 StackNode* frame) {
+    if (ptr == nullptr) {
+      assert(size == 0);
+      return;
+    }
     auto res = alloc_set.emplace(Allocation(std::move(name), ptr, size, frame));
     assert(res.second);
     total_size += size;
   }
   void deallocate(std::string&& name, const void* ptr, std::uint64_t size,
                   StackNode* frame) {
+    if (ptr == nullptr) {
+      assert(size == 0);
+      return;
+    }
     auto key = Allocation(std::move(name), ptr, size, frame);
     auto it  = alloc_set.find(key);
     if (it == alloc_set.end()) {