MinDalle_StableDiff/Python39/Lib/test/mp_fork_bomb.py

19 lines
448 B
Python
Raw Normal View History

2022-09-17 15:26:13 +03:00
import multiprocessing, sys
def foo():
print("123")
# Because "if __name__ == '__main__'" is missing this will not work
# correctly on Windows. However, we should get a RuntimeError rather
# than the Windows equivalent of a fork bomb.
if len(sys.argv) > 1:
multiprocessing.set_start_method(sys.argv[1])
else:
multiprocessing.set_start_method('spawn')
p = multiprocessing.Process(target=foo)
p.start()
p.join()
sys.exit(p.exitcode)