简单 for 循环的 Numba 性能不佳（Python 3.10）

Question

my code:

from numba import njit
from functools import wraps
import time

def timeit(my_func):
    @wraps(my_func)
    def timed(*args, **kw):
    
        tstart = time.time()
        output = my_func(*args, **kw)
        tend = time.time()
        
        print('"{}" took {:.3f} ms to execute\n'.format(my_func.__name__, (tend - tstart) * 1000))
        return output
    return timed

@timeit
@njit
def calculate_smth(a,b):
    result = 0
    for i_a in range(a):
        for i_b in range(b):
            result = result + i_a + i_b
    return result

if __name__ == "__main__":
    value = calculate_smth(1000,1000)

without the numba decorator my function completes in ~62ms, with njit decorates (after compiling beforehand) it needs ~370ms. Can someone explain what I am missing?

Answer 1

JIT stands for Just-In-Time - meaning the code is compiled at execution time - as opposed to AOT - Ahead Of Time. As you can read in Numba docs , by default compilation is lazy, ie. it happens on the first function execution in a program.

It also supports AOT compilation, as described here

Another option would be cache=True parameter passed to numba.njit decorator.