python - 启用/禁用 numba JIT 编译的标志？

Question

有谁知道是否有办法(在代码中)禁用/启用 numba JIT 标签以进行调试？

目前，我在 IDE 中使用选择和替换将它们全部注释掉，但是，必须有一种方法可以自动执行此操作。

我已经尝试过使用 nopython 标志，例如:

USE_NOPYTHON = False

@numba.jit(nopython=USE_NOPYTHON)
...

这不会禁用 jit...

目前手动执行此操作:

@numba.jit(nopython=True)
def foo():
    ...

# -->

# @numba.jit(nopython=True)
def foo():
    ...

但是，如果这样的事情是可能的(当然没有丑陋的 IF 语句)，我会很高兴:

USE_JIT = False

if USE_JIT:
    @numba.jit(nopython=True)
    def foo():
        ...
else:
    def foo():
        ...

Answer 1

找到我的答案:

1. pip 安装 pyyaml

2. 您需要在通常调用脚本的目录中创建一个名为“.numba_config.yaml”的文件。

3. 在 yaml 文件中，将键“DISABLE_JIT”设置为您希望的值(true/false)。

您可以使用许多其他标志，这是我的引用:https://numba.pydata.org/numba-doc/dev/reference/envvars.html#environment-variables

注意:当您的 .yaml 文件名为“.numba_config.yaml”时，如果您要离开引用链接中找到的列表，则无需在标志名称前面添加“NUMBA_”。