Aquila-Consortium/csiborgtools
0
0
Fork 0
mirror of https://github.com/Richard-Sti/csiborgtools_public.git synced 2025-05-14 06:31:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

New matches (#69)

Browse source 
* Remove old file

* Add velocity plotting

* add smooth scale

* Fix bug

* Improve paths

* Edit plotting

* Add smoothed density

* Update boundaries

* Add basics

* Further docs

* Remove blank

* Better catalog broadcasting

* Update high res size

* Update plotting routines

* Update routine

* Update plotting

* Fix field saving name

* Add better colormap for environemnt
This commit is contained in:
Richard Stiskalek 2023-06-17 19:52:26 +01:00 • committed by GitHub
parent 73687fd8cc
commit 35ccfb5c67
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 343 additions and 169 deletions
Download patch file Download diff file Expand all files Collapse all files

23
scripts/utils.py
View file

@ -165,6 +165,8 @@ def open_catalogues(args, config, paths, comm):
if args.verbose and rank == 0:
print(f"{datetime.now()}: opening catalogues.", flush=True)
# We first load all catalogues on the zeroth rank and broadcast their
# names.
if rank == 0:
cats = {}
if args.simname == "csiborg":
@ -182,12 +184,27 @@ def open_catalogues(args, config, paths, comm):
name = paths.quijote_fiducial_nsim(nsim, nobs)
cat = ref_cat.pick_fiducial_observer(nobs, rmax=args.Rmax)
cats.update({name: cat})
names = list(cats.keys())
if nproc > 1:
for i in range(1, nproc):
comm.send(cats, dest=i, tag=nproc + i)
comm.send(names, dest=i, tag=nproc + i)
else:
cats = comm.recv(source=0, tag=nproc + rank)
names = comm.recv(source=0, tag=nproc + rank)
comm.Barrier()
# We then broadcast the catalogues to all ranks, one-by-one as MPI can
# only pass messages smaller than 2GB.
if nproc == 1:
return cats
if rank > 0:
cats = {}
for name in names:
if rank == 0:
for i in range(1, nproc):
comm.send(cats[name], dest=i, tag=nproc + i)
else:
cats.update({name: comm.recv(source=0, tag=nproc + rank)})
return cats