From b45621b9394c1a8ae58b37bf760b05739cbaea70 Mon Sep 17 00:00:00 2001
From: Jean-Matthieu Etancelin <jean-matthieu.etancelin@univ-reims.fr>
Date: Wed, 15 Oct 2014 14:19:40 +0200
Subject: [PATCH] Fix profiling of communication kernels

---
 HySoP/hysop/gpu/multi_gpu_particle_advection.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/HySoP/hysop/gpu/multi_gpu_particle_advection.py b/HySoP/hysop/gpu/multi_gpu_particle_advection.py
index c71965f77..7ecc839c8 100644
--- a/HySoP/hysop/gpu/multi_gpu_particle_advection.py
+++ b/HySoP/hysop/gpu/multi_gpu_particle_advection.py
@@ -837,11 +837,15 @@ class MultiGPUParticleAdvection(GPUParticleAdvection):
     def get_profiling_info(self):
         for k in [self.copy, self.transpose_xy, self.transpose_xy_r,
                   self.transpose_xz, self.transpose_xz_r,
-                  self.num_advec_and_remesh,
-                  self.num_advec, self.num_remesh, self.num_remesh_comm_l,
-                  self.num_remesh_comm_r, self.num_advec_and_remesh,
-                  self.num_advec_and_remesh_comm_r,
-                  self.num_advec_and_remesh_comm_l]:
+                  self.num_advec,
+                  self.num_remesh,
+                  self.num_advec_and_remesh]:
             if k is not None:
                 for p in k.profile:
                     self.profiler += p
+        if self._is2kernel:
+            self.profiler += self.num_remesh_comm_l.profile[0]
+            self.profiler += self.num_remesh_comm_r.profile[0]
+        else:
+            self.profiler += self.num_advec_and_remesh_comm_l.profile[0]
+            self.profiler += self.num_advec_and_remesh_comm_r.profile[0]
-- 
GitLab