torch_mlu/csrc/aten/core/caching_allocator.cpp
uint64_t currentMemoryCached(int device_id) {
  retriveDebugFlag();
  if (debug_mode) {
    return debugging_allocator.get_memory_stats_for_device(device_id).cached_size;
  } else {
    return caching_allocator.get_memory_stats_for_device(device_id).cached_size;
  }
}
// return the current memory cached on MLU
return the current memory cached on MLU
return the current memory cached on MLU
uint64_t currentMemoryCached(int device_id) {
  retriveDebugFlag();
  if (debug_mode) {
    return debugging_allocator.get_memory_stats_for_device(device_id).cached_size;
  } else {
    return caching_allocator.get_memory_stats_for_device(device_id).cached_size;
  }
}
uint64_t currentMemoryCached ( int device_id ) { retriveDebugFlag ( ) ; if ( debug_mode ) { return debugging_allocator . get_memory_stats_for_device ( device_id ) . cached_size ; } else { return caching_allocator . get_memory_stats_for_device ( device_id ) . cached_size ; } }
return the current memory cached on MLU
return the current memory cached on MLU