From ac9f6dc6ae89dad9418d668f73d28d42f23eda96 Mon Sep 17 00:00:00 2001
From: Matthieu Schaller <matthieu.schaller@durham.ac.uk>
Date: Sat, 14 Jul 2018 18:06:36 +0200
Subject: [PATCH] Link the gravity M-M tasks to their cells.

---
 src/engine.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/engine.c b/src/engine.c
index f22eacf8ea..c946477470 100644
--- a/src/engine.c
+++ b/src/engine.c
@@ -2637,6 +2637,13 @@ void engine_count_and_link_tasks_mapper(void *map_data, int num_elements,
       if (t->subtype == task_subtype_external_grav) {
         error("Found a sub-pair/external-gravity task...");
       }
+
+      /* Link M-M tasks to cells */
+    } else if (t->type == task_type_grav_mm) {
+      atomic_inc(&ci->nr_tasks);
+      atomic_inc(&cj->nr_tasks);
+      engine_addlink(e, &ci->grav, t);
+      engine_addlink(e, &cj->grav, t);
     }
   }
 }
-- 
GitLab