def do_integration(): cfg = swarmng.config(nsys=64, nbod=3, log_writer='bdb', log_output_db=output_file_name, log_interval=1, integrator='hermite_cpu_log', time_step=0.01, nogpu=1) ref = swarmng.generate_ensemble(cfg) ens = ref.clone() swarmng.init(cfg) integ = swarmng.Integrator.create(cfg) integ.ensemble = ens swarmng.sync for i in range(1, final_time + 1): integ.destination_time = i integ.integrate()
def do_integration(): cfg = swarmng.config( nsys=64, nbod=3, log_writer='bdb', log_output_db=output_file_name, log_interval=1, integrator='hermite_cpu_log', time_step=0.01, nogpu=1 ); ref = swarmng.generate_ensemble( cfg ) ens = ref.clone() swarmng.init(cfg) integ = swarmng.Integrator.create( cfg ) integ.ensemble = ens swarmng.sync for i in range(1,final_time+1): integ.destination_time = i integ.integrate()
#!/usr/bin/env python2 # -*- coding: utf8 -*- import swarmng import time for c in range(0,2): swarmng.init(swarmng.config(CUDA_DEVICE=c,verbose=1)) integ = swarmng.Integrator.create(swarmng.config(integrator="hermite",time_step=0.001)) times = [] for nb in range(3,7): integ.ensemble = swarmng.generate_ensemble(swarmng.config(nsys=8000,nbod=nb)) integ.destination_time = 10.0 start = time.clock() integ.integrate() times.append(time.clock() - start) print(times)