Running nested functions using numba

Question

I am trying to use numba recently to speedup parts of my code in python. I was trying to run function 1 from inside function 2 while they are both compiled with numba but it is not working. Here is my code:

import numba as nb
from math import acos
from time import time

@nb.jit("void()")
def myfunc():
    s = 0
    for i in range(10000000):
        s += acos(0.5)
    print('The sum is: ', s)


@nb.jit("void()")
def myfunc2():
    myfunc()


tic = time()
myfunc2()
toc = time()
print(toc-tic)

When I call myfunc() the code works and I get a result much faster than when I am not using numba. However, when I call the myfunc2 I see this error:

 File "~/.spyder-py3/temp.py", line 22, in <module>
    myfunc2()

RuntimeError: missing Environment

Anybody has any ideas why calling a function from insdie another is not working in this case?

Answer 1

Numba v0.39+

A fix was introduced in v0.39. As per the Release Notes:

PR #2986: Fix environment propagation

See github pull #2986 for more details.

Numba pre-v0.39

This is a known issue. As described in github issue #2411:

Seems like the environment pointer is not passed properly across nopython functions.

Amending as below to remove print() from numba functions should fix this:

import numba as nb
from math import acos
from time import time

@nb.jit("void()")
def myfunc():
    s = 0
    for i in range(10000000):
        s += acos(0.5)
    return s

@nb.jit("void()")
def myfunc2():
    return myfunc()

tic = time()
x = myfunc2()  # 10471975.511390356
toc = time()
print(toc-tic)