Specialize topology in benchmark scripts.

fast/diffusion_3D_wBCs.py

@@ -42,7 +42,12 @@
 
 print("dx %s, dy %s, dz %s" % (dx, dy, dz))
 
-grid = Grid(shape=(nx, ny, nz), extent=(2., 2., 2.))
+topology = None
+#xDSL only handles 2D domain decomposition right now
+if args.xdsl:
+    topology = ('*', '*', 1)
+
+grid = Grid(shape=(nx, ny, nz), extent=(2., 2., 2.), topology=topology)
 u = TimeFunction(name='u', grid=grid, space_order=so)
 
 a = Constant(name='a')

fast/mpi.sh

@@ -41,7 +41,8 @@ do
 
       # To uncomment to check what's going on without capturing the output.
       # echo OMP_NUM_THREADS=$threads
-       mpirun -np $MPI_NUM_RANKS --bind-to=core python $bench -so $so --devito 1
+      #  mpirun -np $MPI_NUM_RANKS --bind-to=core python $bench -so $so --devito 1
+      #  mpirun -np $MPI_NUM_RANKS --bind-to=core python $bench -so $so --xdsl 1
 
       # Get the runtimes
       devito_time=$(get_runtime mpirun -np $MPI_NUM_RANKS --bind-to=core python $bench -so $so --devito 1)

fast/wave3d_b.py

@@ -45,7 +45,13 @@
 origin = as_tuple(0.0 for _ in range(len(shape)))  # What is the location of the top left corner.
 domain_size = tuple((d-1) * s for d, s in zip(shape, spacing))
 extent = np.load("so%s_grid_extent%s.npz" % (so, shape_str))['arr_0']
-grid = Grid(shape=shape, extent=as_tuple(extent))
+
+topology = None
+#xDSL only handles 2D domain decomposition right now
+if args.xdsl:
+    topology = ('*', '*', 1)
+
+grid = Grid(shape=(nx, ny, nz), extent=(2., 2., 2.), topology=topology)
 
 # With the velocity and model size defined, we can create the seismic model that
 # encapsulates this properties. We also define the size of the absorbing layer as