tree_d_3d.kernels.log_quadtree_moments.log_quadtree_moments_metal

The Metal version (Apple-silicon GPU, float32) of the log_quadtree_moments kernel.

log_quadtree_moments_pyccel.py next to this file is the CPU version with the same arguments. One thread per node; only the first node_count rows of the outputs are written. See :mod:tree_d_3d.kernels._metal.

Functions

Name	Description
log_quadtree_moments	Complex multipole moments a_0 ... a_p of every node.

log_quadtree_momentsfunction#

View source

def log_quadtree_moments(pos: Array, charge: Array, perm: Array, node_count: int, node_center: Array, node_start: Array, node_end: Array, node_qabs: Array, moment_re: Array, moment_im: Array) -> None

Complex multipole moments a_0 … a_p of every node.