We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7c277e6 commit 9c52e77Copy full SHA for 9c52e77
sumpy/fmm.py
@@ -179,10 +179,17 @@ def p2p(self):
179
exclude_self=self.exclude_self,
180
strength_usage=self.strength_usage)
181
182
- @memoize_method
183
def opencl_fft_app(self, shape, dtype, inverse):
184
- with cl.CommandQueue(self.cl_context) as queue:
185
- return get_opencl_fft_app(queue, shape, dtype, inverse)
+ from pytools import memoize_in
+
+ @memoize_in(self.cl_context, (
186
+ SumpyTreeIndependentDataForWrangler.opencl_fft_app,
187
+ shape, dtype, inverse))
188
+ def app():
189
+ with cl.CommandQueue(self.cl_context) as queue:
190
+ return get_opencl_fft_app(queue, shape, dtype, inverse)
191
192
+ return app()
193
194
# }}}
195
0 commit comments