gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-05 07:02:38 24 4
gpt4 key购买 nike

我想编写一个函数,它既可以作为 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'"

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

最佳答案

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

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

关于python - Numba:回退到对象模式时抑制错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63509051/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com