-
Notifications
You must be signed in to change notification settings - Fork 16
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
Implement distributed layer potential evaluation #224
base: main
Are you sure you want to change the base?
Conversation
global_traversal: Any | ||
centers: Any | ||
expansion_radii: Any | ||
global_qbx_centers: Any | ||
qbx_center_to_target_box: Any | ||
non_qbx_box_target_lists: Any | ||
center_to_tree_targets: Any |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Eventually: improve type annotation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Add a 2D and a 3D test.
pytential/qbx/distributed.py
Outdated
fmm_backend=fmm_backend, | ||
**kwargs) | ||
else: | ||
self._use_target_specific_qbx = _use_target_specific_qbx |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Document that this is an internal code path for workers.
@@ -8,7 +8,7 @@ git+https://github.com/inducer/modepy.git#egg=modepy | |||
git+https://github.com/inducer/pyopencl.git#egg=pyopencl | |||
git+https://github.com/inducer/islpy.git#egg=islpy | |||
git+https://github.com/inducer/loopy.git#egg=loopy | |||
git+https://github.com/inducer/boxtree.git#egg=boxtree | |||
git+https://github.com/gaohao95/boxtree.git@dist-pytential#egg=boxtree |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This needs to be reverted back after inducer/boxtree#93 lands.
#239 should get rid of the mypy problems. |
This PR implements the distributed layer potential evaluation using MPI. On the high level, it broadcasts the necessary geometry data from the root rank to the worker rank, and each worker rank constructs a local copy of the geometry data independently.
❗ Depends on inducer/boxtree#93.