(完全解决）(numba)TypingError: Failed in nopython mode pipeline( Invalid use of Function)

音程

已于 2022-03-03 19:44:57 修改

阅读量1.8w

点赞数 8

分类专栏： python 文章标签： python

于 2022-03-03 19:21:10 首次发布

本文链接：https://blog.csdn.net/qq_43391414/article/details/123261500

版权

python 专栏收录该内容

72 篇文章

订阅专栏

本文介绍了在使用Numba库尝试加速Python代码时遇到的TypingError问题，具体表现为Numba无法理解`str`函数和高级索引操作。作者建议并非所有代码都适合加速，特别是涉及字符串操作和复杂索引的情况。解决方案包括避免使用`nopython=True`选项或寻找替代的非Numba优化方法。此外，文章强调了理解Numba对代码优化的限制以及选择合适加速场景的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用numba加速的时候报错：

from numba import jit
@jit(nopython=True)
def nodets2key(batch: int, node: int, ts: float):
    key = '-'.join([str(batch), str(node), str(ts)])
    return key

在pycharm中报错如下：

numba.errors.TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Invalid use of Function(<class 'str'>) with argument(s) of type(s): (int64)

解决办法：

根据那个Invalid use of Function后面括号里面的就是报错的来源，因此，我们定位到str函数。

也就是说，numba不知道str是一个什么东西，无法进行优化，加速失败。

类似地还有，比如下面这个也会报错：

@jit(nopython=True)
def foo(a):
    indices = np.where(a>0.5)
    a[indices] = 1. / a[indices]#numba不知道这个高级索引a[indices]怎么用。
    return a

所以我的观点是：

1.不要用加速了，并不是什么代码都可以和适合加速的，加速大家应该知道适合于那些循环多以及大量基本数据类型数学计算的地方。如果什么函数都用加速，报错报死你，不信你试一试，这就是一个例子。

2.将(nopython=True)去掉，退化为普通加速，就不会报错了。

参考：https://stackoverflow.com/questions/61885520/numba-invalid-use-of-function-with-arguments-of-types。