Changeset 2984

Timestamp:

May 26, 2006, 3:15:44 PM (18 years ago)

Author:

ole

Message:

Parallel version of stochastic study

Location:

development/stochastic_study

Files:

• project.py (modified) (1 diff)
• run_model.py (modified) (6 diffs)

development/stochastic_study/project.py

@@ -27 +27 @@
 
 # Stats (Suresh ?)
-number_of_realisations = 2
+number_of_realisations = 16000
 #std_dev = 0.0026  #Range is 26.035 cm
 #std_dev = 0.0013  #Range is 26.035 cm

development/stochastic_study/run_model.py

@@ -32 +32 @@
 from pyvolution.util import file_function
 from caching.caching import cache
+import pypar
 
 # Application specific imports
@@ -37 +38 @@
 
 
+#-----------------------------------------------------------------------------
+# Read in processor information
+#-----------------------------------------------------------------------------
+
+myid = pypar.rank()
+numprocs = pypar.size()
+processor_name = pypar.Get_processor_name()
+
+print 'I am process %d of %d running on %s' %(myid, numprocs, processor_name)
 
 
@@ -42 +52 @@
 # Setup computational domain
 #----------------------------------------------------------------------------- 
-print 'Creating domain from', project.mesh_filename
+#print 'Creating domain from', project.mesh_filename
 
 domain = Domain(project.working_dir + project.mesh_filename,
                 use_cache=False,
-                verbose=True)                
+                verbose=False)                
                 
 
-print 'Number of triangles = ', len(domain)
-print domain.statistics()
+#print 'Number of triangles = ', len(domain)
+#print domain.statistics()
 
 
@@ -61 +71 @@
 #------------------------------------------------------------------------------
 
-function = file_function(project.boundary_filename, domain, verbose = True)
+function = file_function(project.boundary_filename, domain, verbose = False)
 Bts = Transmissive_Momentum_Set_Stage_boundary(domain, function) #Input wave
 Br = Reflective_boundary(domain) #Wall
@@ -84 +94 @@
 for filename in os.listdir(project.working_dir):
     if filename.startswith(project.basename) and filename.endswith('.pck'):
-        print 'Reading %s' %filename
+        #print 'Reading %s' %filename
         fid = open(filename)
         V = cPickle.load(fid)
@@ -91 +101 @@
         # For each column (each realisation)
         for i in range(V.shape[1]):
-            if i == 0:
-                name = project.basename + '_original'
-            else:
-                name = project.basename                
-            domain.set_name(name)                     #Output name
-            domain.set_quantity('elevation', V[:,i])  #Assign bathymetry
-            domain.set_time(0.0)                      #Reset time
 
-            #---------------------------------------------------
-            # Evolve system through time
-            #---------------------------------------------------
-            print 'Running realisation %d of %d in block %s'\
-                  %(i, V.shape[1], filename)
-            print 'Z Range:', min(V[:,i]), max(V[:,i])
+            # Distribute work in round-robin fashion
+            if i%numprocs == myid:
+                
+                name = project.basename + '_P%d' %myid    
+                domain.set_name(name)                     #Output name
+                domain.set_quantity('elevation', V[:,i])  #Assign bathymetry
+                domain.set_time(0.0)                      #Reset time
 
-            t0 = time.time()
-            for t in domain.evolve(yieldstep = timestep,
-                                   finaltime = finaltime):
-                domain.write_time()
-                
+                #---------------------------------------------------
+                # Evolve system through time
+                #---------------------------------------------------
+                print 'P%d: Running realisation %d of %d in block %s'\
+                      %(myid, realisation, V.shape[1], filename)
+
+                t0 = time.time()
+                for t in domain.evolve(yieldstep = timestep,
+                                       finaltime = finaltime):
+                    pass
+                    #domain.write_time()
+                    
         
-            print 'Simulation of realisation %d took %.2f seconds'\
-                  %(realisation, time.time()-t0)
+                print 'P%d: Simulation of realisation %d took %.2f seconds'\
+                      %(myid, realisation, time.time()-t0)
 
 
 
 
-            #---------------------------------------------------
-            # Now extract the 3 timeseries (Ch 5-7-9) and store them
-            # in three files for this realisation
-            f = file_function(project.working_dir + domain.filename + '.sww',
-                              quantities='stage',
-                              interpolation_points=project.gauges,
-                              verbose=True)
+                #---------------------------------------------------
+                # Now extract the 3 timeseries (Ch 5-7-9) and store them
+                # in three files for this realisation
+                f = file_function(project.working_dir + domain.filename + '.sww',
+                                  quantities='stage',
+                                  interpolation_points=project.gauges,
+                                  verbose=False)
 
 
-            simulation_name = project.working_dir + \
-                              domain.filename + '_realisation_%d' %realisation
+                simulation_name = project.working_dir + \
+                                  project.basename + '_realisation_%d' %realisation
 
-            for k, name in enumerate(project.gauge_names):
-                fid = open(simulation_name + '_' + name + '.txt', 'w')
-                for t in f.get_time():
-                    #For all precomputed timesteps
-                    val = f(t, point_id = k)[0]
-                    fid.write('%f %f\n' %(t, val))
+                for k, name in enumerate(project.gauge_names):
+                    fid = open(simulation_name + '_' + name + '.txt', 'w')
+                    for t in f.get_time():
+                        #For all precomputed timesteps
+                        val = f(t, point_id = k)[0]
+                        fid.write('%f %f\n' %(t, val))
 
-                fid.close()
+                    fid.close()
 
                     
 
             realisation += 1            
+
+
+pypar.finalize()