예제 #1
0
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
예제 #2
0
 def setUp(self):
     np.random.seed(0)
     bm.use_fftw()
     pass