run_cluster_sim.py

# This is meant to be run, for example, with the command mpirun -n 4 -python run_cluster_sim.py
# the code above will make 4 different processes


import sys
import numpy as np
from SimRunner import BitFlipRunner,FredkinRunner
import math

# always include these lines of MPI code!
from mpi4py import MPI
comm = MPI.COMM_WORLD

size = comm.Get_size()  # number of MPI procs
rank = comm.Get_rank()  # i.d. for local proc


#defaults = {'localization':18., 'location':.5, 'depth':3, 'tilt':2., 'beta':1., 'tau':1., 'scale':1., 'dt':1/10000, #'lambda':1, 'N':10_000}

initial_params = {'N':50_000,'dt':1/5_000,'target_work':None, 'k':1, 'depth':8, 'location':2}

#this will be sued to geenrate names for the different sim outputs dynamically
def save_func(self):
   lmda = int(1000*self.params['lambda'])
   N = int(self.params['N'])
   k = int(self.params['k'])
   return [f'k{k}lmda{lmda:03d}N{N}', None]

simrun = BitFlipRunner()
simrun.save_name = save_func
simrun.change_params(initial_params)

# This code is to make a save name manually if you dont want to use simrun.save_name
#run_name = 'bf_N{initial_params['N']/'
#save_dir = f'/home/kylejray/FQ_sims/results/{run_name}/'



#max_L = 8
#L_lists = [ L_range[i*max_L:(i+1)*max_L] for i in range(math.ceil(len(L_range)/max_L))]



# my_param = params[rank]

#dt = [1/item for item in [500, 1_000, 5_000, 10_000, 15_000, 20_000, 30_000]][rank]
#n = [item*1000 for item in [1,4,16,64,256,512]][rank]

'''
locals = [ 20, 50, 100, 200]
zs = [ 2, 3, 4, 5, 6]
combos=[]
for l in locals:
   for z in zs:
      loc = z/np.sqrt(2*l)
      combos.append({'localization':l, 'location':loc})

print(len(combos))
c_lists = [ combos[i*size:(i+1)*size] for i in range(math.ceil(len(combos)/size))]

for combos in c_lists:
   params = combos[rank]
'''

# generating a list of parameter dictionaries for the different combos we want to try
p_list=[]
p_keys = ['k', 'lambda']
lmdas = np.linspace(0, .2, 8)
ks = [1,4]
for k in ks:
   for l in lmdas:
      p_list.append({key:val for key,val in zip(p_keys,[k,l])})

#separates the different parameter combinations into different lists, depending on how many processes you allow
p_lists = [ p_list[i*size:(i+1)*size] for i in range(math.ceil(len(p_list)/size))]


for params in p_lists:
   # perform local computation within each rank
   simrun.change_params(params[rank])

   # save parameters to output file by printing, this might not work unless using SLURM on Demon
   sys.stdout.write('my rank:{} of {}, params={} '.format(rank+1, size, params[rank]))
   
   simrun.run_sim(verbose=True)
   
   # save your results
   simrun.save_sim()