1

ある 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'

何が間違っている可能性がありますか?

4

0 に答える 0