diff options
| -rwxr-xr-x | astro/oskar/run_oskar.py | 24 |
1 files changed, 19 insertions, 5 deletions
diff --git a/astro/oskar/run_oskar.py b/astro/oskar/run_oskar.py index 744dda3..79c2518 100755 --- a/astro/oskar/run_oskar.py +++ b/astro/oskar/run_oskar.py @@ -19,8 +19,9 @@ from time import time def run_oskar(configfile, model, freq, vis_ms, vis_oskar=None, - telescope=None, chunksize=None, use_gpus=None, - gpu_ids=None, num_devices=None, dryrun=False): + telescope=None, chunksize=None, double_precision=None, + use_gpus=None, gpu_ids=None, num_devices=None, + dryrun=False): """ Update simulation settings in the configuration file, and run the simulator ``oskar_sim_interferometer``. @@ -34,6 +35,7 @@ def run_oskar(configfile, model, freq, vis_ms, vis_oskar=None, simulator = "oskar_sim_interferometer" for item, value in [ ("simulator/max_sources_per_chunk", chunksize), + ("simulator/double_precision", double_precision), ("simulator/use_gpus", use_gpus), ("simulator/cuda_device_ids", gpu_ids), ("simulator/num_devices", num_devices), @@ -95,10 +97,15 @@ def main(): help="force to only use GPUs") exgrp1.add_argument("--use-cpu", dest="use_cpu", action="store_true", help="force to only use CPUs") - exgrp2 = parser.add_mutually_exclusive_group(required=True) - exgrp2.add_argument("-l", "--list", dest="listfile", + exgrp2 = parser.add_mutually_exclusive_group() + exgrp2.add_argument("--use-double", dest="use_double", action="store_true", + help="use double-precision float numbers") + exgrp2.add_argument("--use-single", dest="use_single", action="store_true", + help="use single-precision float numbers") + exgrp3 = parser.add_mutually_exclusive_group(required=True) + exgrp3.add_argument("-l", "--list", dest="listfile", help="list of frequencies [MHz] and input sky models") - exgrp2.add_argument("-i", "--items", dest="items", nargs="+", + exgrp3.add_argument("-i", "--items", dest="items", nargs="+", help="list of sky model items in format of " + "'<freq[MHz]>:<skymodel-file>'") args = parser.parse_args() @@ -110,6 +117,12 @@ def main(): elif args.use_cpu: use_gpus = False + double_precision = None + if args.use_double: + double_precision = True + elif args.use_single: + double_precision = False + if not os.path.exists(args.outdir): os.mkdir(args.outdir) print("Created output visibility directory: %s" % args.outdir) @@ -147,6 +160,7 @@ def main(): run_oskar(configfile=configfile, freq=freq, model=skyfile, vis_ms=vis_ms, vis_oskar=vis_oskar, telescope=args.telescope, chunksize=args.chunksize, + double_precision=double_precision, use_gpus=use_gpus, gpu_ids=args.gpu_ids, num_devices=args.num_devices, dryrun=args.dryrun) |