如何使用 nopython true 检查推断的 numba 类型？

Question

没有

nopython=True

numba.typeof

报告变量类型。

如何使用

nopython=True

检查变量的类型？

import numba

@numba.jit(nopython=True)
def f():
    a = 0
    print(numba.typeof(a))
f()

给予

UntypedAttributeError: Unknown attribute 'typeof' of type Module(<module 'numba'>)

我知道虽然可以做到

inspect_types()

，但是对于复杂的功能来说不太方便。

Answer 1

if type(v) is numba.float32: ...

Answer 2

注释掉装饰器并重新运行代码，然后你应该看到 numba 的类型