#!/usr/bin/bpftrace

BEGIN
{
    printf("Tracing kmem allocation stacks (kmalloc, kmem_cache_alloc). ");
    printf("Hit Ctrl-C to end.\n");
}

tracepoint:kmem:kmalloc,
tracepoint:kmem:kmem_cache_alloc
{
    @bytes[kstack(5), comm] = stats(args->bytes_alloc);
}
