diff --git a/sumpy/p2p.py b/sumpy/p2p.py index 76bf36a5ae40550adf8ea32bd76d833cb61a9895..aa9da6eea0d838d43a73c715ac5c4b23b95ea837 100644 --- a/sumpy/p2p.py +++ b/sumpy/p2p.py @@ -581,7 +581,7 @@ class P2PFromCSR(P2PBase): """] + [""" <> is_self = (isrc == target_to_source[itgt]) """ if self.exclude_self else ""] - + [f"<> strength_{i} = strengths[{i}, isrc]" for + + [f"<> strength_{i} = strength[{i}, isrc]" for i in set(self.strength_usage)] + loopy_insns + [f"""