From a955fe9b175ebad5fddd2a1c4f4cf1d33483e424 Mon Sep 17 00:00:00 2001
From: Rafael Ravedutti <rafaelravedutti@gmail.com>
Date: Fri, 5 Jan 2024 17:52:39 +0100
Subject: [PATCH] Fix minor problem with cell reference
Signed-off-by: Rafael Ravedutti <rafaelravedutti@gmail.com>
---
src/pairs/sim/interaction.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pairs/sim/interaction.py b/src/pairs/sim/interaction.py
index 5502d85..34bfa58 100644
--- a/src/pairs/sim/interaction.py
+++ b/src/pairs/sim/interaction.py
@@ -66,10 +66,10 @@ class NeighborFor:
cell_neighbors = self.cell_lists.cell_neighbors
for shape in self.shapes:
- start = sum([cell_nneighs[cell][s] for s in range(shape)], 0)
# FIXME: Without the inline, the 'cell' expression is being generated after
# its usage in the loop upper limit
cell = ScalarOp.inline(particle_cell[self.particle])
+ start = sum([cell_nneighs[cell][s] for s in range(shape)], 0)
for k in For(self.sim, start, start + cell_nneighs[cell][shape]):
particle_id = cell_neighbors[cell][k]
--
GitLab