Как изнутри функции Python узнать, выполняется ли она в графическом процессоре с помощью Numba или регулярно вызывается ⇐ Python

[Гость]

У меня есть функция, которая иногда вызывается с помощью Numba как функции устройства для выполнения на графическом процессоре, а иногда я вызываю ее непосредственно из обычного Python на хосте:

Код: Выделить всё

def process():
# perform computation

process_cuda = cuda.jit(device=True)(process)

sometimes I call

Код: Выделить всё

process()

from directly from Python, and sometimes I invoke the

Код: Выделить всё

process_cuda

wrapper as a kernel with Numba.
My question, how can I tell, from within the

Код: Выделить всё

process

function, if it was called directly from Python or if it is executing as a Numba device function?


Источник: https://stackoverflow.com/questions/766 ... in-the-gpu