tree_d_3d.kernels.pair_energy
The pair_energy kernel of the Barnes-Hut and FMM solvers: Pyccel, NumPy and CUDA versions with the same arguments.
kernel(...) runs the CUDA version when the arrays are CuPy arrays, and otherwise the
Pyccel version, or the NumPy version if Pyccel is unavailable (see
cunumpy.kernels.set_host_kernel_implementation to choose).
Attributes
| Name | Description |
|---|---|
kernel | No description. |
Modules
pair_energy_metalThe Metal version (Apple-silicon GPU, float32) of the pair_energy kernel.pair_energy_numpyThe NumPy version of the pair_energy kernel, for machines without Pyccel.pair_energy_pyccelThe Pyccel version (CPU) of the pair_energy kernel of the Barnes-Hut and FMM solvers.
kernelattributemodule attribute#
kernel = xp.kernels.Kernel.from_folder(__name__, **KERNEL_OPTIONS)