Changeset a42fec0 in sasmodels for compare.py


Ignore:
Timestamp:
Aug 4, 2014 3:20:07 PM (10 years ago)
Author:
HMP1 <helen.park@…>
Branches:
master, core_shell_microgels, costrafo411, magnetic_model, release_v0.94, release_v0.95, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests
Children:
8cdb9f1
Parents:
099e053
Message:

Speed-up of 3X, compare.py working

File:
1 edited

Legend:

Unmodified
Added
Removed
  • compare.py

    r473183c ra42fec0  
    44import datetime 
    55 
    6 from Models.sasmodel import SasModel, load_data, set_beam_stop, plot_data 
     6from sasmodel import SasModel, load_data, set_beam_stop, plot_data 
    77 
    88 
     
    4646    if len(sys.argv) > 1: 
    4747        N = int(sys.argv[1]) 
    48     data = load_data('JUN03289.DAT') 
     48    data = load_data('December/DEC07098.DAT') 
    4949    set_beam_stop(data, 0.004) 
    5050 
     51    dtype = "float" 
    5152    pars = dict( 
    52         scale=1, radius=64.1, length=266.96, sldCyl=.291e-6, sldSolv=5.77e-6, background=0, 
    53         cyl_theta=0, cyl_phi=0, radius_pd=0.1, radius_pd_n=10, radius_pd_nsigma=3,length_pd=0.1, 
    54         length_pd_n=5, length_pd_nsigma=3, cyl_theta_pd=0.1, cyl_theta_pd_n=5, cyl_theta_pd_nsigma=3, 
     53        scale=.08, radius=64.1, length=266.96, sldCyl=.291e-6, sldSolv=5.77e-6, background=.1, 
     54        cyl_theta=0, cyl_phi=0, 
     55        radius_pd=0.1, radius_pd_n=10, radius_pd_nsigma=3, 
     56        length_pd=0.1,length_pd_n=5, length_pd_nsigma=3, 
     57        cyl_theta_pd=0.1, cyl_theta_pd_n=5, cyl_theta_pd_nsigma=3, 
    5558        cyl_phi_pd=0.1, cyl_phi_pd_n=10, cyl_phi_pd_nsigma=3, 
    5659        ) 
     
    6265    cpu_time = toc()*1000./N 
    6366 
    64     from code_cylinder import GpuCylinder 
     67 
     68    from Models.code_cylinder import GpuCylinder 
    6569    model = SasModel(data, GpuCylinder, dtype='f', **pars) 
    6670    tic() 
     
    6973    gpu_time = toc()*1000./N 
    7074 
     75    print "gpu/cpu", max(gpu/cpu) 
     76    cpu *= max(gpu/cpu) 
    7177    relerr = (gpu - cpu)/cpu 
    7278    print "max(|(ocl-omp)/ocl|)", max(abs(relerr)) 
     
    7682    plt.subplot(131); plot_data(data, cpu); plt.title("omp t=%.1f ms"%cpu_time) 
    7783    plt.subplot(132); plot_data(data, gpu); plt.title("ocl t=%.1f ms"%gpu_time) 
    78     plt.subplot(133); plot_data(data, 1e8*relerr); plt.title("relerr x 10^8"); plt.colorbar() 
     84    plt.subplot(133); plot_data(data, relerr); plt.title("relerr x 10^8"); plt.colorbar() 
    7985    plt.show() 
    8086 
     
    8591    if len(sys.argv) > 1: 
    8692        N = int(sys.argv[1]) 
    87     data = load_data('DEC07133.DAT') 
     93    data = load_data('DEC07106.DAT') 
    8894    set_beam_stop(data, 0.004) 
    8995 
     
    99105    cpu_time = toc()*1000./N 
    100106 
    101     from code_ellipse import GpuEllipse 
     107    from Models.code_ellipse import GpuEllipse 
    102108    model = SasModel(data, GpuEllipse, dtype='f', **pars) 
    103109    tic() 
     
    116122    plt.show() 
    117123 
    118 def coreshell(N=4): 
     124def coreshell(N=1): 
    119125    import sys 
    120126    import matplotlib.pyplot as plt 
     
    122128    if len(sys.argv) > 1: 
    123129        N = int(sys.argv[1]) 
    124     data = load_data('DEC07133.DAT') 
     130    data = load_data('December/DEC07098.DAT') 
    125131    set_beam_stop(data, 0.004) 
    126132 
     
    141147    cpu_time = toc()*1000./N 
    142148 
    143     from code_coreshellcyl import GpuCoreShellCylinder 
     149    from Models.code_coreshellcyl import GpuCoreShellCylinder 
    144150    model = SasModel(data, GpuCoreShellCylinder, dtype='f', **pars) 
    145151    tic() 
Note: See TracChangeset for help on using the changeset viewer.