bench: Patch jacobian operators + MPI (see issue #1744)

devitocodes · Sep 3, 2021 · 55e4c46 · 55e4c46
1 parent 47f3ac3
commit 55e4c46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/user/benchmark.py b/benchmarks/user/benchmark.py
@@ -66,7 +66,7 @@ def run_op(solver, operator, **options):
         dm = solver.model.dm
         # Because sometime dm is zero, artificially add a non zero slice
         if dm.data.min() == 0 and dm.data.max() == 0:
-            dm.data[..., np.min([25, dm.shape[-1]//4])] = .1
+            dm.data[..., np.min([25, dm.shape_global[-1]//4])] = .1
         return op(dm, **options)
     elif operator == "jacobian_adjoint":
         # I think we want the forward + gradient call, need to merge retvals