diff --git a/src/cudamatrix/cu-allocator.h b/src/cudamatrix/cu-allocator.h index 3028be2b576..d7d65da806a 100644 --- a/src/cudamatrix/cu-allocator.h +++ b/src/cudamatrix/cu-allocator.h @@ -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 within the cache during current execution + size_t GetMaxAllocatedMemory() { return max_allocated_memory_; } + CuMemoryAllocator(); // Allows you to set options: must be called before any Malloc function is