Ipyparallel и Numba (Jit) - PullRequest
Новая
       21

Ipyparallel и Numba (Jit)

0 голосов
/ 09 сентября 2018

Я пытался запустить код, используя ipyparallel. Тем не менее, так как некоторые из моих функций проходят через JIT. В итоге консоль ipython ничего не делает и не зависает. Вот пример: 

import time
import numpy
from numba import jit

from ipyparallel import Client
client = Client()  # create client and direct view to all engines available
dview = client[:]
dview.block = True


with dview.sync_imports():
    import numpy

@jit
def TestingFnc(x):
    a=x[0]
    b=x[1]
    c=x[2]
    Result=0
    for i in range(100000):
        Result = Result + ((a+i)**2 + (b-i)**3) /(c+i)**3
    return numpy.array([Result,1])

d=numpy.array([[0.,0.,0.],[0.,0.,1.],[0.,0.,2]])
Results = dview.map(TestingFnc, d)

Без @jit код выполняется параллельно.

...