multiprocess_generator.py

import wavefunction_generator as wfg
import multiprocessing as mp
import time

start=time.time() #starts timing
simulations = 20 # Number of simulations to run (NOTE: Each successive simulation takes longer to run, 20 simulations ~15mins for 8 logical processors)


if __name__ == "__main__":  # Allows for the safe importing of the main module
    """
    Assigns child processes (wavefunction_generator.run()) to multiprocessing pool 
    for multithreaded performance increase
    """
    print("There are {} logical processors on this machine".format(mp.cpu_count()))
    print("Running {} simulations...".format(simulations))
    
    max_processes = mp.cpu_count()  # number of simultaneous processes cannot excede the number of logical processors
    pool = mp.Pool(max_processes)
    processes = pool.map_async(wfg.run, range(1,simulations+1)) # Assign processes to the processing pool
    pool.close() #finish assigning
    pool.join() #begins multiprocessing

    print("Time elapsed = {}s".format(round(time.time()-start,2)))  #finishes timing
    print("\a")