python-3.x - numba jitted函数内的冲洗标准输出

Question

显然numba不支持sys.stdout.flush和print("", flush=True)。

什么是清除jitted函数中的“打印”的好方法？

Answer 1

您可以使用 objmode() context manager来使用numba的nopython JIT模式不支持的代码:

import numba

@numba.njit
def f(x):
    x *= 2
    with numba.objmode():
        print(x, flush=True)
    return x + 1

print(f'f(7) = {f(7)}')
# 14
# f(7) = 15

如文档中所述，由于所涉及的开销，仅应在关键性能代码段之外使用。
注意:当问题最初于2017年提出时，我认为此功能不可用。