args = parse() n_iterations = n_iterations if args['turns'] == None else args['turns'] n_particles = n_particles if args['particles'] == None else args['particles'] n_bunches = n_bunches if args['bunches'] == None else args['bunches'] n_turns_reduce = n_turns_reduce if args['reduce'] == None else args['reduce'] seed = seed if args['seed'] == None else args['seed'] approx = args['approx'] timing.mode = args['time'] os.environ['OMP_NUM_THREADS'] = str(args['omp']) withtp = bool(args['withtp']) precision = args['precision'] bm.use_precision(precision) bm.use_mpi() bm.use_fftw() worker.assignGPUs(num_gpus=args['gpu']) worker.greet() if worker.isMaster: worker.print_version() worker.initLog(bool(args['log']), args['logdir']) worker.initTrace(bool(args['trace']), args['tracefile']) worker.taskparallelism = withtp mpiprint(args) # initialize simulation
def setUp(self): np.random.seed(0) bm.use_fftw() pass