@@ -122,12 +122,11 @@ class PointPotentialSource(_SumpyP2PMixin, PotentialSource):
122122 def __init__ (self , nodes , * ,
123123 fmm_order : Optional [int ] = False ,
124124 fmm_level_to_order : Optional [Union [bool , Callable [..., int ]]] = None ,
125- expansion_factory : Optional [DefaultExpansionFactory ] \
125+ expansion_factory : Optional [DefaultExpansionFactory ]
126126 = default_expansion_factory ,
127127 tree_build_kwargs : Optional [Mapping ] = None ,
128128 trav_build_kwargs : Optional [Mapping ] = None ,
129- setup_actx : Optional [ArrayContext ] = None
130- ):
129+ setup_actx : Optional [ArrayContext ] = None ):
131130 """
132131 :arg nodes: The point potential source given as a
133132 :class:`pyopencl.array.Array`
@@ -136,7 +135,8 @@ def __init__(self, nodes, *,
136135 If both arguments are not given a direct point-to-point calculation
137136 is used.
138137 :arg fmm_level_to_order: An optional callable that returns the FMM order
139- to use for a given level. Mutually exclusive with *fmm_order* argument.
138+ to use for a given level. Mutually exclusive with *fmm_order*
139+ argument.
140140 :arg expansion_factory: An expansion factory to get the expansion objects
141141 when an FMM is used.
142142 :arg tree_build_kwargs: Keyword arguments to be passed when building the
@@ -262,11 +262,11 @@ def _get_tree(self, target_discr):
262262 @memoize_method
263263 def _get_exec_insn_func (self , source_kernels , target_kernels , target_discr ):
264264 if self .fmm_level_to_order is False :
265- sources = self ._nodes
266- targets = flatten (target_discr .nodes (), self ._setup_actx , leaf_class = DOFArray )
267265 def exec_insn (actx , strengths , kernel_args , dtype , return_timing_data ):
266+ sources = self ._nodes
267+ targets = flatten (target_discr .nodes (), actx , leaf_class = DOFArray )
268268 p2p = self .get_p2p (actx , source_kernels = source_kernels ,
269- target_kernels = target_kernels )
269+ target_kernels = target_kernels )
270270
271271 evt , output = p2p (actx .queue ,
272272 targets = targets ,
0 commit comments