tree_d_3d.kernels.direct_sum.direct_sum_metal

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

direct_sum_pyccel.py next to this file is the CPU version with the same arguments. One thread per target; see :mod:tree_d_3d.kernels._metal.

Functions

Name	Description
direct_sum	All-pairs Coulomb potential and field on the GPU; same conventions as the CPU.

direct_sumfunction#

View source

def direct_sum(targets: Array, exclude: Array, pos: Array, charge: Array, softening: float, phi: Array, field: Array, status: Array) -> None

All-pairs Coulomb potential and field on the GPU; same conventions as the CPU.