Доступ к глобальным переменным в Numba Jeded Parallel Function ⇐ Python

[Anonymous]

Обычно оочетные функции Numba не могут динамически получить доступ к глобальным переменным - значение фиксируется на время компиляции. Питоническое поведение может быть вызвано путем принуждения objmode . (Существует снижение производительности для использования objmode ).

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

from numba import njit, objmode
@njit
def access_global_variable_1():
return global_variable

@njit
def access_global_variable_2():
with objmode(retval='int64'):
retval = globals()['global_variable']
return retval

global_variable = 0
access_global_variable_1() # returns 0
access_global_variable_2() # returns 0

global_variable = 1
access_global_variable_1() # returns 0
access_global_variable_2() # returns 1
< /code>
Тем не менее, это ломается, когда мы параллелируем функции < /p>
from numba import njit, objmode, prange
import numpy as np

@njit
def _add(a):
with objmode():
globals()['acc'] += a

@njit(parallel=True)
def parallel_sum_global(arr):
for i in prange(len(arr)):
_add(arr[i])

@njit(parallel=False)
def sum_global(arr):
for i in prange(len(arr)):
_add(arr[i])

@njit(parallel=True)
def parallel_sum_local(arr):
acc = 0
for i in prange(len(arr)):
acc += arr[i]
return acc

n = 100
print('True answer:', np.arange(n).sum()) # True answer: 4950

acc = 0
parallel_sum_global(np.arange(n))
print('Numba parallel global answer:', acc) # Numba parallel answer: 78

acc = 0
sum_global(np.arange(n))
print('Numba global answer:', acc) # Numba global answer: 4950

acc = parallel_sum_local(np.arange(n))
print('Numba parallel local answer:', acc) # Numba parallel local answer: 4950

Это не может быть проблемой, связанной с доступом к глобальным переменным, поскольку sum_global работает как задумано. Это не может быть проблемой, связанной с параллелизацией, так как Sum_parallel_local работает так, как предполагалось. Это только проблема, которая возникает, когда доступ к глобальным переменным и параллелизированию. Есть ли обходные пути?>

Подробнее здесь: https://stackoverflow.com/questions/795 ... l-function