Changeset 977 for codes/icosagcm/devel/Python/test/py/partition.py
- Timestamp:
- 09/26/19 18:49:01 (5 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
codes/icosagcm/devel/Python/test/py/partition.py
r825 r977 3 3 from dynamico import meshes 4 4 from dynamico import parallel 5 from dynamico.dev import unstructured as unst6 5 from dynamico import maps 7 from dynamico .parmetis import partition_graph6 from dynamico import partition 8 7 9 8 from mpi4py import MPI … … 11 10 mpi_rank, mpi_size = comm.Get_rank(), comm.Get_size() 12 11 print '%d/%d starting'%(mpi_rank,mpi_size) 13 14 #from dynamico import partition15 12 16 13 import math as math … … 50 47 part = 0*xadj_loc[0:-1]; 51 48 52 # unst.ker.dynamico_partition_graph(mpi_rank, mpi_size, vtxdist, xadj_loc, adjncy_loc, 4, part) 53 partition_graph(comm, vtxdist, xadj_loc, adjncy_loc, part, nparts=4) 49 partition.partition_graph(comm, vtxdist, xadj_loc, adjncy_loc, part, nparts=4) 54 50 55 51 for i in range(len(part)): … … 66 62 mydegree, mybounds = [get_mycells(x) for x in primal_deg, primal_vertex] 67 63 print '%d : len(mydegree)=%d'%(mpi_rank, len(mydegree)) 68 vertex_list = sorted(set( unst.list_stencil(mydegree,mybounds)))64 vertex_list = sorted(set(partition.list_stencil(mydegree,mybounds))) 69 65 print '%d : len(vertex_list))=%d'%(mpi_rank, len(vertex_list)) 70 66 get_myvertices = parallel.Get_Indices(dim_vertex, vertex_list) … … 116 112 plt.xlim(-190.,190.) 117 113 plt.ylim(-90.,90.) 118 plt.savefig('fig_partition/A%03d.p df'%mpi_rank, dpi=1600)114 plt.savefig('fig_partition/A%03d.png'%mpi_rank, dpi=160) 119 115 120 116 fig, ax = plt.subplots() … … 124 120 plt.xlim(-190.,190.) 125 121 plt.ylim(-90.,90.) 126 plt.savefig('fig_partition/B%03d.p df'%mpi_rank, dpi=1600)122 plt.savefig('fig_partition/B%03d.png'%mpi_rank, dpi=160)
Note: See TracChangeset
for help on using the changeset viewer.