Skip to content
Merged
Changes from 7 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions src/cudamatrix/cu-allocator.h
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,12 @@ class CuMemoryAllocator {

void PrintMemoryUsage() const;

// returns the current memory allocated within the cache
size_t GetAllocatedMemory() { return allocated_memory_; }

// returns the maximum memory used witnin the cache during current execution
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

typo in comment - witnin

size_t GetMaxAllocatedMemory() { return max_allocated_memory_; }

CuMemoryAllocator();

// Allows you to set options: must be called before any Malloc function is
Expand Down