arrayred = thetarun[rank * sizered:(rank + 1) * sizered] freepar = Grid.freepar print("lenrun, sizered", lenrun, sizered) allPlin = [] allPloop = [] for i, theta in enumerate(arrayred): parameters = copy.deepcopy(Grid.parref) truetheta = Grid.valueref + theta * Grid.delta idx = nrun * lenrun + rank * sizered + i print("nrun, rank, i", nrun, rank, i) parameters["PathToOutput"] = os.path.join( OUTPATH, 'output' + str(nrun) + str(rank) + str(i)) for k, var in enumerate(freepar): parameters[var] = truetheta[k] # parameters['h'] = 1/parameters['invh'] Plin, Ploop = Grid.CompPterms(parameters) idxcol = np.full([Plin.shape[0], 1], idx) allPlin.append(np.hstack([Plin, idxcol])) allPloop.append(np.hstack([Ploop, idxcol])) if (i == 0) or ((i + 1) % 100 == 0): print("theta check: ", Grid.flattenedgrid[idx], theta, truetheta) # np.save(os.path.join(outpk, "temp", "Plin_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPlin)) # np.save(os.path.join(outpk, "temp", "Ploop_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPloop)) np.save( os.path.join(outpk, "Plin_run%s_rank%s.npy" % (str(nrun), str(rank))), np.array(allPlin)) np.save( os.path.join(outpk, "Ploop_run%s_rank%s.npy" % (str(nrun), str(rank))), np.array(allPloop))
allPlin = [] allPloop = [] for i, theta in enumerate(arrayred): parameters = copy.deepcopy(Grid.parref) truetheta = Grid.valueref + theta * Grid.delta #idx = nrun * lenrun + rank * sizered + i #print("nrun, rank, i", nrun, rank, i) idx = i print ("i on tot", i, sizearray) parameters["PathToOutput"] = os.path.join(OUTPATH, 'output' + str(nrun) + str(rank) + str(i)) for k, var in enumerate(freepar): parameters[var] = truetheta[k] # parameters['h'] = 1/parameters['invh'] kin, Plin, z, Omega_m = Grid.CompPterms(parameters) bird = pybird.Bird(kin, Plin, Omega_m, z, full=False) nonlinear.PsCf(bird) bird.setPsCfl() resum.Ps(bird, full=False) bird.subtractShotNoise() Plin, Ploop = bird.formatTaylor() idxcol = np.full([Plin.shape[0], 1], idx) allPlin.append(np.hstack([Plin, idxcol])) allPloop.append(np.hstack([Ploop, idxcol])) if (i == 0) or ((i+1) % 100 == 0): print("theta check: ", Grid.flattenedgrid[idx], theta, truetheta) # np.save(os.path.join(outpk, "temp", "Plin_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPlin)) # np.save(os.path.join(outpk, "temp", "Ploop_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPloop)) np.save(os.path.join(outpk, "Plin_run%s_rank%s.npy" % (str(nrun), str(rank))), np.array(allPlin))