numba - 使用 NumbaPro 从另一个 GPU 内核调用一个 GPU 内核

Question

我希望从一个 GPU 内核调用另一个：

import numpy
from numbapro import vectorize

sig = 'int16(int16, int16)' 

@vectorize([sig], device=True, target='gpu')
def sum(a, b):
    return a + b

@vectorize([sig], target='gpu')
def proxy(a, b):
    return sum(a, b)

result = proxy(5, 10) # this will fail!

我已经device=True在被调用函数上添加了，但它似乎没有起到作用。

失败的行产生此错误：TypingError: Untyped global name 'sum'

可能有什么问题？