Skip to content

Commit 8b4ae18

Browse files
committed
add OutputRecord.free_events(), OutputRecord.peak_free_events()
1 parent bc1b799 commit 8b4ae18

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

mp_reader/malloc_stats.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -616,6 +616,15 @@ def peak_usage(self) -> tuple[int, int, int]:
616616

617617
return int(eid), int(alloc_counts[eid]), int(byte_counts[eid])
618618

619+
def free_events(self) -> list[OutputEvent]:
620+
return [e for e in self.event_table if e.type == EventType.FREE]
621+
622+
def peak_free_events(self) -> list[OutputEvent]:
623+
"""
624+
Return free events that correspond to allocations available at
625+
peak usage
626+
"""
627+
return self.pseudo_frees_at_time(self.peak_usage()[0])
619628

620629
@dataclass
621630
class ObjectTree:

0 commit comments

Comments
 (0)