Skip to content
Projects

tree_d_3d.kernels.soften_pairs

The soften_pairs 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

NameDescription
kernelNo description.

Modules

  • soften_pairs_metalThe Metal version (Apple-silicon GPU, float32) of the soften_pairs kernel.
  • soften_pairs_numpyThe NumPy version of the soften_pairs kernel, for machines without Pyccel.
  • soften_pairs_pyccelThe Pyccel version (CPU) of the soften_pairs kernel of the Barnes-Hut and FMM solvers.

kernelattributemodule attribute#

kernel = xp.kernels.Kernel.from_folder(__name__, **KERNEL_OPTIONS)