examples: Add kwargs to operators and wavesolvers

examples/seismic/elastic/operators.py

@@ -50,8 +50,10 @@ def ForwardOperator(model, geometry, space_order=4, save=False, **kwargs):
     """
 
     v = VectorTimeFunction(name='v', grid=model.grid,
+                           save=geometry.nt if save else None,
                            space_order=space_order, time_order=1)
     tau = TensorTimeFunction(name='tau', grid=model.grid,
+                             save=geometry.nt if save else None,
                              space_order=space_order, time_order=1)
 
     lam, mu, b = model.lam, model.mu, model.b
@@ -63,6 +65,7 @@ def ForwardOperator(model, geometry, space_order=4, save=False, **kwargs):
              model.damp * dt * mu * (grad(v.forward) + grad(v.forward).T))
 
     srcrec = src_rec(v, tau, model, geometry)
-    op = Operator([u_v] + [u_t] + srcrec, subs=model.spacing_map, name="ForwardElastic")
+    op = Operator([u_v] + [u_t] + srcrec, subs=model.spacing_map, name="ForwardElastic",
+                  **kwargs)
     # Substitute spacing terms to reduce flops
     return op

examples/seismic/tti/tti_example.py

@@ -8,7 +8,8 @@
 
 
 def tti_setup(shape=(50, 50, 50), spacing=(20.0, 20.0, 20.0), tn=250.0,
-              space_order=4, nbl=10, preset='layers-tti', **kwargs):
+              kernel='centered', space_order=4, nbl=10, preset='layers-tti',
+              **kwargs):
 
     # Two layer model for true velocity
     model = demo_model(preset, shape=shape, spacing=spacing,
@@ -17,15 +18,15 @@ def tti_setup(shape=(50, 50, 50), spacing=(20.0, 20.0, 20.0), tn=250.0,
     # Source and receiver geometries
     geometry = setup_geometry(model, tn)
 
-    return AnisotropicWaveSolver(model, geometry, space_order=space_order)
+    return AnisotropicWaveSolver(model, geometry, space_order=space_order, **kwargs)
 
 
 def run(shape=(50, 50, 50), spacing=(20.0, 20.0, 20.0), tn=250.0,
         autotune=False, time_order=2, space_order=4, nbl=10,
         kernel='centered', full_run=False, **kwargs):
 
     solver = tti_setup(shape=shape, spacing=spacing, tn=tn, space_order=space_order,
-                       nbl=nbl, **kwargs)
+                       nbl=nbl, kernel=kernel, **kwargs)
     info("Applying Forward")
 
     rec, u, v, summary = solver.forward(autotune=autotune, kernel=kernel)