ある GPU カーネルから別の 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'
何が間違っている可能性がありますか?