Как сделать гонку между мультипроцессорами в python
У меня есть функция, которая учитывает число. Это зависит от некоторого случайного условия.
Итак, что я пытаюсь сделать, так это запустить несколько процессоров в этой функции, и процессор, который сначала находит фактор, возвращает значение, и все процессоры завершают работу.
То, что у меня есть до сих пор, очень неправильно. Процессоры не завершаются, и я также не знаю, как получить значение, которое было возвращено функцией
flag = False
def rho(n, processor):
while True:
x = random.randrange(1, n-1)
x2 = x
gcd = 1
c = random.randrange(1, n-1)
while gcd == 1:
x = (x**2 + c) % n
x2 = (x2**2 + c) % n
x2 = (x2**2 + c) % n
gcd = math.gcd(abs(x - x2), n)
if gcd != n:
flag = True
print("Factor was found from "+process+" and is ", gcd)
return gcd
if __name__ == "__main__":
p1 = multiprocessing.Process(target=rho, args=(91, "process 1" ))
p2 = multiprocessing.Process(target=rho, args=(91, "process 2"))
p1.start()
p2.start()
if flag:
p1.terminate()
p2.terminate()
Результат:
Factor was found from process 2 and is 13
Factor was found from process 1 and is 7
Ответы 1
Вы можете использовать multiprocessing.Pool и его методы map(), imap_unordered() и т. д. Они также будут возвращать значения из рабочих функций.
Пример (я использовал time.sleep() для имитации трудоемких вычислений):
from time import sleep
from multiprocessing import Pool
def rho(params):
n, processor = params
# your computation here
# ...
sleep(n)
print("Factor was found from " + processor + " and is 42")
return 42
if __name__ == "__main__":
with Pool() as pool:
for result in pool.imap_unordered(
rho, ((10, "process 1"), (1, "process 2"))
):
print("Result I got:", result)
break # <-- I don't want other results, so break
Отпечатки:
Factor was found from process 2 and is 42
Result I got: 42
Обновлено: две разные функции:
from time import sleep
from multiprocessing import Pool
def fn1(n, p):
sleep(n)
print("Factor was found from " + p + " and is 42")
return 42
def fn2(n, p):
sleep(n)
print("Factor was found from " + p + " and is 99")
return 99
def rho(params):
what_to_call, n, processor = params
return what_to_call(n, processor)
if __name__ == "__main__":
with Pool() as pool:
for result in pool.imap_unordered(
rho, ((fn1, 10, "process 1"), (fn2, 1, "process 2"))
):
print("Result I got:", result)
break # <-- I don't want other results, so break
@userpr Я использовал sleep(n) для имитации тяжелых вычислений. Конечно, в вашем коде удалите sleep()
Если я возьму сон (n), он снова сработает, иначе проблема может заключаться в том, что он работает, но занимает слишком много времени.
@userpr Я пытался разделить вычисления на несколько процессов. Но если ваш алгоритм экспоненциальный или хуже, то даже многопроцессорность вам не поможет.
Могу ли я использовать другую функцию в пуле? Итак, прямо сейчас, что вызывает ту же функцию. Но могу ли я запустить процессор для функции 1, а затем другой для функции 2 и остановиться, как только один из них что-то вернет?
@userpr Я обновил свой ответ, как вызвать пул с двумя разными функциями.
Когда я добавляю это, мой код перестает работать. И почему спать(n)?