tree_d_3d.kernels.direct_sum
The direct_sum 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
direct_sum_metalThe Metal version (Apple-silicon GPU, float32) of the direct_sum kernel.direct_sum_numpyThe NumPy version of the direct_sum kernel, for machines without Pyccel.direct_sum_pyccelThe Pyccel version (CPU) of the direct_sum kernel of the Barnes-Hut and FMM solvers.
kernelattributemodule attribute#
kernel = xp.kernels.Kernel.from_folder(__name__, **KERNEL_OPTIONS)