tree_d_3d.kernels.quadtree_moments.quadtree_moments_pyccel
The Pyccel version (CPU) of the quadtree_moments kernel of the Barnes-Hut and FMM solvers.
quadtree_moments_cuda.cu next to this file is the CUDA version (GPU) and quadtree_moments_numpy.py the
version for machines without Pyccel, all with the same arguments; the folder’s
__init__.py declares the kernel (see :mod:tree_d_3d.kernels). The flat
quadtree is described in :mod:tree_d_3d.flat_tree.
Functions
| Name | Description |
|---|---|
quadtree_moments | Signed moments of every node about its centre; node_s is xx, xy, yy. |
quadtree_momentsfunction#
def quadtree_moments(pos: float[:, :], charge: float[:], perm: int64[:], node_count: int, node_center: float[:, :], node_start: int64[:], node_end: int64[:], node_q: float[:], node_qabs: float[:], node_p: float[:, :], node_s: float[:, :]) -> NoneSigned moments of every node about its centre; node_s is xx, xy, yy.