函数返回 jitclass 的 Numba jit() 函数签名

Question

如果我定义一个jitclass:

SPEC = [ ... ]
@jitclass
class Foo:
    ...

如何为返回 class 实例的 jit-ed 函数指定类型签名。当我尝试时：

@jit("Foo(float32[:])")
def some_function(a: np.ndarray): ...

我得到 NameError: name 'Foo' is not defined，即使它在上面声明。

Answer 1

来自

尝试以下操作：

@jit(Foo.class_type.instance_type(float32[:]))
def some_function(a: np.ndarray): ...

Numba jit() function signature for function returning jitclass