numba.jit 将函数和签名作为输入

Question

我的目标：

我有一个函数，我想看看它对 numba 选项/参数列表的执行情况（nopython、no gil、parallel 等和 Signature！）

所以我想做类似的事情：

def foo(a):
   return a*2


for signature in list:
   foo_jit = numba.jit(foo, signature)
   
   print("speed for jitted func given signature")

遗憾的是，这引发了一个错误，因为 @numba.jit() 作为装饰器采用签名作为第一个输入，而 numba.jit() 作为函数采用函数作为输入。我已经尝试在 jit 函数之前分配签名，但是你不能 jit 一个函数两次。

Answer 1

为了测试这种方法，我会尝试在循环中定义一个函数：

for signature in list:

    @jit(signature)
    def foo(a):
       return a*2

    start = time.time()
    foo(whatever_data_you_have)
    end = time.time()
    print("Elapsed (with compilation) = %s" % (end - start))

您当然可以阅读in the guide 第一次是编译，因此，如果您想要不编译的时间，请再添加一次测试块，这将从缓存中加载并且不会计算经过的编译时间。