diff --git a/examples/seismic/elastic/operators.py b/examples/seismic/elastic/operators.py index fe077194b9..b504de7a9c 100644 --- a/examples/seismic/elastic/operators.py +++ b/examples/seismic/elastic/operators.py @@ -63,6 +63,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 diff --git a/examples/seismic/tti/tti_example.py b/examples/seismic/tti/tti_example.py index c0bbc207b8..02d8077ecb 100644 --- a/examples/seismic/tti/tti_example.py +++ b/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,7 +18,7 @@ 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, @@ -25,7 +26,7 @@ def run(shape=(50, 50, 50), spacing=(20.0, 20.0, 20.0), tn=250.0, 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)