Andreas Klöckner authored 8 months ago and Andreas Klöckner committed 8 months ago

Also, required from __future__ import annotations

* Support M2L with FFT on ToyContext

* flake8 fixes

* Fix src_rscale, tgt_rscale ordering

* More tests with FFT

* Use sumpy.toys in test_translations

* flake8 fixes

* Optimize M2L for GPU

* Move icoeff_tgt to top level iname

* Fix substitution

* use loopy branch

* remove unused imports

* go back to loopy main

* Reduce diff

* move all optimizations to m2l_translation

* Remove extraneous FIXME

Co-authored-by: Andreas Klöckner <inform@tiker.net>

* use loopy fft

* fix inverse

* Implement broadcasting FFT

* use enum for fft backend

* Add gh-129 link

* unit test for loopy_fft

* Unit test for loopy_fft and fix warnings

* don't use vkfft only if x86 mac

* Add missing import

* Fix platform.machine()

* Use a separate class for M2L translation

* Fix docs and caching

* Fix p2p warning

* Use VkFFT for M2L generate data

* Fix profiling events

* simplify m2l data zeros

* Add pyvkfft to requirements

* Fix flake8 warning

* Fix typo

* VkFFT for M2L preprocess local

* vkfft for postprocess local

* Fix AggregateProfilingEvent

* Fix another typo

* M2L Translation Factory

* vim markers

* Fix tests

* Fix toys

* Fix test_m2l_toeplitz

* Fix more tests

* Use a better rscale to get the test passing

* Use pytential dev branch

* remove whitespace on blank line

* Try 2r/order instead of r/order

* fix using updated pytential

* Fix tests

* use pytential branch with pyvkfft req

* Add explanation about caller being responsible for the FFT

* Fix for bessel

* Add pyvkfft to setup.py reqs

* use list comprehension

* Type annotations

* fix vim marker

* remove unused function

* m2l_data_inner -> m2l_data

* more descriptive name for child_knl

* knl -> expr_knl for clarity

* move loop unroll to optimized

* Add explanation about translation_classes_dependent_data_loopy_knl

* make coeffs output only and rewrite

* Re-arrange m2l so that event processing is easier

* flake8: single quotes -> double quotes

* Fix data not being input

* make args to cached_vkfft_app explicit

* cache vkfftapp in wrangler

* keep coeffs is_input and is_output for e2e

* out-of-place fft

* Use a separate queue for configuration

* allocate array for out-of-place

* fix typo

* Remove caching of opencl fft app

* Comment out pytentual fork

* fix vkfft queues

* use private API for now

* Add comment on pyvkfft PR

* remove inplace

Co-authored-by: Andreas Klöckner <inform@tiker.net>

* 4r/order for biharmonic

* Add a test for magnitudes

* Use a separate class for M2L translation

* Fix docs and caching

* M2L Translation Factory

* vim markers

* Fix tests

* Fix toys

* Fix test_m2l_toeplitz

* Fix more tests

* Use a better rscale to get the test passing

* Use pytential dev branch

* Try 2r/order instead of r/order

* Compute translation_classes_data if not given

* Remove SumpyTranslationClassesData

* update tests

* use readable syntax

* Fix HelmholtzDerivativeTaker

* Use 2, 3, 4 for 2D

* Divide by r**2 at the end

* Write the recursion as one expression