Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

FFT using pyvkfft and use loopy callables #114

Merged
merged 67 commits into from
Aug 1, 2022
Merged
Changes from 1 commit
Commits
Show all changes
67 commits
Select commit Hold shift + click to select a range
ad39de6
Use a separate class for M2L translation
isuruf Apr 29, 2022
9cb4caa
Fix docs and caching
isuruf Apr 29, 2022
c5bc9b2
Fix p2p warning
isuruf Apr 29, 2022
88c9fd8
Use VkFFT for M2L generate data
isuruf May 1, 2022
bab8f38
Fix profiling events
isuruf May 1, 2022
68864a2
simplify m2l data zeros
isuruf May 1, 2022
03cc94e
Add pyvkfft to requirements
isuruf May 2, 2022
c440950
Fix flake8 warning
isuruf May 2, 2022
78faead
Fix typo
isuruf May 2, 2022
4f33038
VkFFT for M2L preprocess local
isuruf May 2, 2022
efb6d99
vkfft for postprocess local
isuruf May 2, 2022
2e1b10e
Fix AggregateProfilingEvent
isuruf May 2, 2022
59d7be6
Fix another typo
isuruf May 2, 2022
72b4875
M2L Translation Factory
isuruf May 3, 2022
83f7fd8
vim markers
isuruf May 3, 2022
bed782f
Merge branch 'isuruf/m2l' into fft
isuruf May 3, 2022
7cf5404
Fix tests
isuruf May 5, 2022
584d2c9
Fix toys
isuruf May 5, 2022
6f5ad1f
Fix test_m2l_toeplitz
isuruf May 5, 2022
47a4a27
Fix more tests
isuruf May 5, 2022
60ef708
Use a better rscale to get the test passing
isuruf May 5, 2022
85e0ed1
Use pytential dev branch
isuruf May 5, 2022
9f74eec
Merge branch 'isuruf/m2l' into fft
isuruf May 5, 2022
9162d17
Merge branch 'main' into m2l
isuruf May 6, 2022
3880661
Merge branch 'isuruf/m2l' of https://github.com/inducer/sumpy into fft
isuruf May 6, 2022
a57f727
remove whitespace on blank line
isuruf May 6, 2022
ea6a99c
Try 2r/order instead of r/order
isuruf May 6, 2022
df991c6
Merge branch 'isuruf/m2l' of https://github.com/inducer/sumpy into fft
isuruf May 8, 2022
4eaad6a
fix using updated pytential
isuruf May 8, 2022
9119880
Merge branch 'main' of https://github.com/inducer/sumpy into fft
isuruf May 13, 2022
21236d7
Fix tests
isuruf May 13, 2022
9ddd3a9
use pytential branch with pyvkfft req
isuruf May 14, 2022
e5dea13
Add explanation about caller being responsible for the FFT
isuruf May 14, 2022
52de95d
Fix for bessel
isuruf May 14, 2022
3ddc0db
Merge branch 'main' into fft
inducer May 19, 2022
bec642e
Add pyvkfft to setup.py reqs
isuruf May 25, 2022
e6d62a4
use list comprehension
isuruf May 25, 2022
774f869
Type annotations
isuruf May 25, 2022
2c8a5bf
fix vim marker
isuruf May 25, 2022
61ddc0b
remove unused function
isuruf May 25, 2022
50f8bb3
m2l_data_inner -> m2l_data
isuruf May 25, 2022
c3eaa32
more descriptive name for child_knl
isuruf May 25, 2022
9cc214e
knl -> expr_knl for clarity
isuruf May 25, 2022
7d9f535
move loop unroll to optimized
isuruf May 25, 2022
07c1c93
Add explanation about translation_classes_dependent_data_loopy_knl
isuruf May 25, 2022
25dd7fc
make coeffs output only and rewrite
isuruf May 25, 2022
1252c5a
Re-arrange m2l so that event processing is easier
isuruf May 25, 2022
3b3f6f3
flake8: single quotes -> double quotes
isuruf May 25, 2022
8e9649f
Fix data not being input
isuruf May 25, 2022
a66a7cc
make args to cached_vkfft_app explicit
isuruf May 25, 2022
fff0d13
cache vkfftapp in wrangler
isuruf May 25, 2022
e80c71d
keep coeffs is_input and is_output for e2e
isuruf May 25, 2022
8d72d66
out-of-place fft
isuruf Jun 6, 2022
98b4bcc
Use a separate queue for configuration
isuruf Jun 6, 2022
d7da927
Merge branch 'main' into fft
isuruf Jun 6, 2022
3e9632b
allocate array for out-of-place
isuruf Jun 6, 2022
aafe8b2
fix typo
isuruf Jun 6, 2022
d4bfc05
Remove caching of opencl fft app
isuruf Jun 6, 2022
899006e
Comment out pytentual fork
isuruf Jun 9, 2022
a0942ed
fix vkfft queues
isuruf Jun 9, 2022
a5996a5
use private API for now
isuruf Jun 22, 2022
da7f310
Merge branch 'main' into fft
isuruf Jun 23, 2022
bf083c9
Merge branch 'main' into fft
isuruf Jun 25, 2022
741a1cc
Add comment on pyvkfft PR
isuruf Jun 25, 2022
15df148
Merge branch 'main' into fft
inducer Jun 25, 2022
b8af2bd
remove inplace
isuruf Aug 1, 2022
e073d73
Merge branch 'main' into fft
isuruf Aug 1, 2022
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion sumpy/fmm.py
Original file line number Diff line number Diff line change
Expand Up @@ -548,7 +548,7 @@ def box_target_list_kwargs(self):
# }}}

def run_opencl_fft(self, queue, input_vec, inverse, wait_for, inplace):
app = self.tree_indep.opencl_fft_app(input_vec.shape, input_vec.dtype,
app = get_opencl_fft_app(queue, input_vec.shape, input_vec.dtype,
isuruf marked this conversation as resolved.
Show resolved Hide resolved
inplace)
return run_opencl_fft(app, queue, input_vec, inverse, wait_for)

Expand Down Expand Up @@ -697,6 +697,8 @@ def multipole_to_local_precompute(self):
m2l_translation_classes_dependent_data_view.shape[0]

if ntranslation_classes == 0:
result.append(pyopencl.array.empty_like(
m2l_translation_classes_dependent_data_view))
continue

data = self.translation_classes_data
Expand Down