python - Numba:回退到对象模式时抑制错误

Question

我想编写一个函数，它既可以作为 jitted 函数运行，也可以作为普通 python 或对象模式 numba 运行，具体取决于 numba 是否能够进行类型推断。我实际上更喜欢普通的 python，但除了编写我自己的装饰器之外似乎没有办法做到这一点。

我为什么要这样做？因为我有一个函数 f 将另一个函数 g 作为参数。所以如果 g 是一个 jitted，那么我希望 f 是 jitted 这样整个事情就可以快速运行，但是如果 g 是普通的 python，然后我希望 f 仍然运行，尽管速度更慢。

简单地用 @jit 注释 f 就可以正常工作，除了我第一次在对象模式下运行该函数时，我收到有关类型推断失败的编译警告.我想取消此警告。

This answer说要使用 @autojit(warn=False) 但不再有任何 autojit。它还说 @jit(warn=False) 会起作用，但这也不起作用，我得到:

KeyError: "<class 'numba.core.cpu.CPUTargetOptions'> does not support option: 'warn'"

必须有一种方法可以在不收到警告的情况下完成此操作，但我似乎无法用谷歌搜索它。感谢您的帮助。

Answer 1

为此有一个环境变量。您可以尝试在运行时设置 NUMBA_DEBUG_TYPEINFER=0。

引用:https://numba.pydata.org/numba-doc/dev/reference/envvars.html?highlight=warn#envvar-NUMBA_DEBUG_TYPEINFER