python numba 指纹错误

Question

我正在尝试使用 numba 优化一些代码。我已经完成了 0.26.0 用户指南 (http://numba.pydata.org/numba-doc/0.26.0/user/jit.html) 中第 1.3.1 节中的初始示例并获得了预期的结果，所以我认为问题不是安装。

这是我的代码：

import numba
import numpy
import random

a = 8
b = 4

def my_function(a, b):
    all_values = numpy.fromiter(range(a), dtype = int)
    my_array = []
    for n in (range(a)):
        some_values = (all_values[all_values != n]).tolist()
        c = random.sample(some_values, b)
        my_array.append(sorted([n] + c))
    return my_array

print(my_function(a, b))

my_function_numba = numba.jit()(my_function)

print(my_function_numba(a, b))

打印出 my_function 调用 returns 的预期结果后，出现以下错误消息：

ValueError                                Traceback (most recent call last)
<ipython-input-8-b5d8983a58f6> in <module>()
     19 my_function_numba = numba.jit()(my_function)
     20 
---> 21 print(my_function_numba(a, b))

ValueError: cannot compute fingerprint of empty list

空表指纹？

Answer 1

我不确定具体的错误，但总的来说，要快速 numba 需要 numpy/python 的特定子集（有关更多信息，请参见 here and here）。所以我可能会这样重写它。

@numba.jit(nopython=True)
def fast_my_function(a, b):
    all_values = np.arange(a)
    my_array = np.empty((a, b + 1), dtype=np.int32)
    for n in range(a):
        some = all_values[all_values != n]
        c = np.empty(b + 1, dtype=np.int32)
        c[1:] = np.random.choice(some, b)
        c[0] = n
        c.sort()
        my_array[n, :] = c
    return my_array

主要注意事项：

没有列表，我正在预先分配所有内容。
不使用生成器（在 python 2 和 3 for n in range(a) 中将转换为快速本机循环）
将 nopython=True 添加到装饰器中，如果我使用无法有效 JIT 的东西，numba 会抱怨。

python numba 指纹错误

python numba fingerprint error

python-3.x

numba