numba_mem.py

from numba import jit
import resource


# Memory Leak Example
def aFun(N):    
    @jit(nopython=True)
    def aLoop(N):
        return N
    return aLoop(N)

M = 100
N = 10
for k in range(M):
    aFun(N)
    if k%10 == 0:
        print 'Memory usage at iteration %i: %s (kb)' % (k,resource.getrusage(resource.RUSAGE_SELF).ru_maxrss)

# Workaround
@jit(nopython=True)
def bLoop(N):
    return N

def bFun(bLoop,N):    
    return bLoop(N)

for k in range(M):
    bFun(bLoop,N)
    if k%10 == 0:
        print 'Memory usage at iteration %i: %s (kb)' % (k,resource.getrusage(resource.RUSAGE_SELF).ru_maxrss)