• F
    Add new APIs for GPU memory monitoring (max_memory_allocated,... · afe02e9d
    From00 提交于
    Add new APIs for GPU memory monitoring (max_memory_allocated, max_memory_reserved, memory_allocated, memory_reserved) (#38657)
    
    * Add new API memory_reserved
    
    * Add memory_allocated, max_memory_reserved and max_memory_allocater
    
    * Fix CI error
    
    * Fix CI error
    
    * Enhance UT
    
    * Add FLAGS_memory_stats_opt
    
    * Add STATS macro functions
    
    * Add StatAllocator
    
    * Fix CI errors
    
    * Add UT
    
    * Fix CI errors
    afe02e9d
stat_allocator.h 1.8 KB