ihk_os_getrusage(): Add per-page-size memory usage accounting
This commit is contained in:
@@ -2,7 +2,10 @@
|
||||
#define __RUSAGE_H
|
||||
|
||||
#include <config.h>
|
||||
#include <page.h>
|
||||
#include <ihk/rusage.h>
|
||||
#include <ihk/atomic.h>
|
||||
#include <memobj.h>
|
||||
|
||||
#ifdef ENABLE_RUSAGE
|
||||
#define RUSAGE_MEM_LIMIT (2 * 1024 * 1024) // 2MB
|
||||
@@ -23,9 +26,9 @@ rusage_rss_add(unsigned long size)
|
||||
unsigned long retval;
|
||||
|
||||
newval = __sync_add_and_fetch(&monitor->rusage_rss_current, size);
|
||||
oldval = monitor->rusage_rss_max;
|
||||
oldval = monitor->rusage_memory_max_usage;
|
||||
while (newval > oldval) {
|
||||
retval = __sync_val_compare_and_swap(&monitor->rusage_rss_max,
|
||||
retval = __sync_val_compare_and_swap(&monitor->rusage_memory_max_usage,
|
||||
oldval, newval);
|
||||
if (retval == oldval) {
|
||||
break;
|
||||
@@ -40,6 +43,88 @@ rusage_rss_sub(unsigned long size)
|
||||
__sync_sub_and_fetch(&monitor->rusage_rss_current, size);
|
||||
}
|
||||
|
||||
static inline void memory_stat_rss_add(unsigned long size, int pgsize)
|
||||
{
|
||||
ihk_atomic_add_long(size, &monitor->rusage_memory_stat_rss[rusage_pgsize_to_pgtype(pgsize)]);
|
||||
}
|
||||
|
||||
static inline void memory_stat_rss_sub(unsigned long size, int pgsize)
|
||||
{
|
||||
ihk_atomic_add_long(-size, &monitor->rusage_memory_stat_rss[rusage_pgsize_to_pgtype(pgsize)]);
|
||||
}
|
||||
|
||||
static inline void rusage_memory_stat_mapped_file_add(unsigned long size, int pgsize)
|
||||
{
|
||||
ihk_atomic_add_long(size, &monitor->rusage_memory_stat_mapped_file[rusage_pgsize_to_pgtype(pgsize)]);
|
||||
}
|
||||
|
||||
static inline void rusage_memory_stat_mapped_file_sub(unsigned long size, int pgsize)
|
||||
{
|
||||
ihk_atomic_add_long(-size, &monitor->rusage_memory_stat_mapped_file[rusage_pgsize_to_pgtype(pgsize)]);
|
||||
}
|
||||
|
||||
static inline int rusage_memory_stat_add(struct vm_range *range, uintptr_t phys, unsigned long size, int pgsize)
|
||||
{
|
||||
/* Is it resident in main memory? */
|
||||
if (range->flag & (VR_REMOTE | VR_IO_NOCACHE | VR_RESERVED)) {
|
||||
return 0;
|
||||
}
|
||||
/* Is it anonymous and pre-paging? */
|
||||
if (!range->memobj) {
|
||||
memory_stat_rss_add(size, pgsize);
|
||||
return 1;
|
||||
}
|
||||
/* Is it devobj or (fileobj and pre-map) or xpmem attachment? */
|
||||
if ((range->memobj->flags & MF_DEV_FILE) ||
|
||||
(range->memobj->flags & MF_PREMAP) ||
|
||||
(range->memobj->flags & MF_XPMEM)
|
||||
) {
|
||||
return 0;
|
||||
}
|
||||
/* Is it anonymous and demand-paging? */
|
||||
if (range->memobj->flags & MF_ZEROOBJ) {
|
||||
memory_stat_rss_add(size, pgsize);
|
||||
return 1;
|
||||
}
|
||||
|
||||
struct page *page = phys_to_page(phys);
|
||||
|
||||
/* Is It file map and cow page? */
|
||||
if ((range->memobj->flags & (MF_DEV_FILE | MF_REG_FILE)) &&
|
||||
!page) {
|
||||
//kprintf("%s: cow,phys=%lx\n", __FUNCTION__, phys);
|
||||
memory_stat_rss_add(size, pgsize);
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* Is it a sharable page? */
|
||||
if (!page) {
|
||||
kprintf("%s: WARNING !page,phys=%lx\n", __FUNCTION__, phys);
|
||||
return 0;
|
||||
}
|
||||
/* Is this the first attempt to map the sharable page? */
|
||||
if(__sync_bool_compare_and_swap(&page->mapped.counter64, 0, 1)) {
|
||||
if(range->memobj->flags & MF_SHM) {
|
||||
memory_stat_rss_add(size, pgsize);
|
||||
} else {
|
||||
rusage_memory_stat_mapped_file_add(size, pgsize);
|
||||
}
|
||||
return 1;
|
||||
} else {
|
||||
return 0;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
static inline void rusage_memory_stat_sub(struct memobj *memobj, unsigned long size, int pgsize)
|
||||
{
|
||||
if(memobj->flags & MF_SHM) {
|
||||
memory_stat_rss_sub(size, pgsize);
|
||||
} else {
|
||||
rusage_memory_stat_mapped_file_sub(size, pgsize);
|
||||
}
|
||||
}
|
||||
|
||||
static inline void
|
||||
rusage_kmem_add(unsigned long size)
|
||||
{
|
||||
@@ -47,11 +132,11 @@ rusage_kmem_add(unsigned long size)
|
||||
unsigned long oldval;
|
||||
unsigned long retval;
|
||||
|
||||
newval = __sync_add_and_fetch(&monitor->rusage_kmem_usage, size);
|
||||
oldval = monitor->rusage_kmem_max_usage;
|
||||
newval = __sync_add_and_fetch(&monitor->rusage_memory_kmem_usage, size);
|
||||
oldval = monitor->rusage_memory_kmem_max_usage;
|
||||
while (newval > oldval) {
|
||||
retval = __sync_val_compare_and_swap(
|
||||
&monitor->rusage_kmem_max_usage,
|
||||
&monitor->rusage_memory_kmem_max_usage,
|
||||
oldval, newval);
|
||||
if (retval == oldval) {
|
||||
break;
|
||||
@@ -63,13 +148,13 @@ rusage_kmem_add(unsigned long size)
|
||||
static inline void
|
||||
rusage_kmem_sub(unsigned long size)
|
||||
{
|
||||
__sync_sub_and_fetch(&monitor->rusage_kmem_usage, size);
|
||||
__sync_sub_and_fetch(&monitor->rusage_memory_kmem_usage, size);
|
||||
}
|
||||
|
||||
static inline void
|
||||
rusage_numa_add(int numa_id, unsigned long size)
|
||||
{
|
||||
__sync_add_and_fetch(monitor->rusage_numa_stat + numa_id, size);
|
||||
__sync_add_and_fetch(monitor->rusage_memory_numa_stat + numa_id, size);
|
||||
rusage_rss_add(size);
|
||||
}
|
||||
|
||||
@@ -77,7 +162,7 @@ static inline void
|
||||
rusage_numa_sub(int numa_id, unsigned long size)
|
||||
{
|
||||
rusage_rss_sub(size);
|
||||
__sync_sub_and_fetch(monitor->rusage_numa_stat + numa_id, size);
|
||||
__sync_sub_and_fetch(monitor->rusage_memory_numa_stat + numa_id, size);
|
||||
}
|
||||
|
||||
static inline void
|
||||
@@ -163,6 +248,31 @@ rusage_rss_sub(unsigned long size)
|
||||
{
|
||||
}
|
||||
|
||||
static inline void memory_stat_rss_add(unsigned long size, size_t pgsize)
|
||||
{
|
||||
}
|
||||
|
||||
static inline void memory_stat_rss_sub(unsigned long size, size_t pgsize)
|
||||
{
|
||||
}
|
||||
|
||||
static inline void rusage_memory_stat_mapped_file_add(unsigned long size, int pgsize)
|
||||
{
|
||||
}
|
||||
|
||||
static inline void rusage_memory_stat_mapped_file_sub(unsigned long size, int pgsize)
|
||||
{
|
||||
}
|
||||
|
||||
static inline int rusage_memory_stat_add(struct vm_range *range, uintptr_t phys, unsigned long size, int pgsize)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
static inline void rusage_memory_stat_sub(struct memobj *memobj, unsigned long size, int pgsize)
|
||||
{
|
||||
}
|
||||
|
||||
static inline void
|
||||
rusage_numa_add(int numa_id, unsigned long size)
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user