Changeset 825 for codes/icosagcm/devel/Python/test/py/partition.py

Timestamp:

04/30/19 18:14:37 (5 years ago)

Author:

dubos

Message:

devel/Python : moved Fortran bindings and *.pyx to dynamico/dev module + necessary changes to test/py/*.py

File:

• codes/icosagcm/devel/Python/test/py/partition.py (modified) (4 diffs)

codes/icosagcm/devel/Python/test/py/partition.py

@@ -1 +1 @@
 print 'Starting'
+
+from dynamico import meshes
+from dynamico import parallel
+from dynamico.dev import unstructured as unst
+from dynamico import maps
+from dynamico.parmetis import partition_graph
 
 from mpi4py import MPI
@@ -7 +13 @@
 
 #from dynamico import partition
-from dynamico import parallel, meshes
-from dynamico import unstructured as unst
 
 import math as math
@@ -46 +50 @@
     part = 0*xadj_loc[0:-1];
 
-    unst.ker.dynamico_partition_graph(mpi_rank, mpi_size, vtxdist, xadj_loc, adjncy_loc, 4, part)
+#    unst.ker.dynamico_partition_graph(mpi_rank, mpi_size, vtxdist, xadj_loc, adjncy_loc, 4, part)
+    partition_graph(comm, vtxdist, xadj_loc, adjncy_loc, part, nparts=4)
 
-#    for i in range(len(part)):
-#        print 'vertex', i+idx_start, 'proc', part[i]
+    for i in range(len(part)):
+        print 'vertex', i+idx_start, 'proc', part[i]
 
 #-----------------------------------------------------------------------------#
@@ -84 +89 @@
 def coriolis(lon,lat): return 0.*lat
 llm, nqdyn, radius = 1,1,1.
-lmesh = meshes.Local_Mesh(pmesh, llm, nqdyn, radius, coriolis)
+planet = maps.SphereMap(radius, 0.)
+lmesh = meshes.Local_Mesh(pmesh, llm, nqdyn, planet)
 
 (primal_deg, primal_vertex, dim_vertex, dim_cell, cell_owner,