aboutsummaryrefslogtreecommitdiffstats
path: root/astro/oskar
diff options
context:
space:
mode:
authorAaron LI <aly@aaronly.me>2018-02-04 14:48:32 +0800
committerAaron LI <aly@aaronly.me>2018-02-04 14:48:32 +0800
commit5084f15e72a9167011657dbe218c7fea066c13dc (patch)
tree00b12f110e5ec05f3719d13c006574f112144941 /astro/oskar
parent25128414d352a209c3e00969bc0bbde601571756 (diff)
downloadatoolbox-5084f15e72a9167011657dbe218c7fea066c13dc.tar.bz2
astro/run_oskar.py: Add --use-double/--use-single arguments
Diffstat (limited to 'astro/oskar')
-rwxr-xr-xastro/oskar/run_oskar.py24
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)