tree_d_3d.kernels.tests.test_metal_fmm_kernels

The Metal versions of the FMM and tree-build kernels, against the CPU versions.

Each test calls a <name>_metal.py function and the Pyccel (or NumPy) version with the same arrays. The GPU computes in float32; skipped without MLX and a Metal GPU.

Attributes

Name	Description
ORDER	No description.
SOFTENING	No description.
pytestmark	No description.

Functions

Name	Description
system	No description.
test_fmm_chain_m2l_downward_evaluate	No description.
test_fmm_traverse_lists_equal_the_cpu_lists	No description.
test_fmm_traverse_reports_a_small_buffer	No description.
test_soften_pairs	No description.
test_soften_pairs_reports_coincident_charges	No description.
test_tree_build_kernels_and_moments	No description.
tree	No description.

ORDERattributemodule attribute#

View source

ORDER = 8

SOFTENINGattributemodule attribute#

View source

SOFTENING = 0.01

pytestmarkattributemodule attribute#

View source

pytestmark = pytest.mark.skipif(not metal_available(), reason='MLX with a Metal GPU is not available')

systemfunction#

View source

def system()

test_fmm_chain_m2l_downward_evaluatefunction#

View source

def test_fmm_chain_m2l_downward_evaluate(system, tree)

test_fmm_traverse_lists_equal_the_cpu_listsfunction#

View source

def test_fmm_traverse_lists_equal_the_cpu_lists(tree)

test_fmm_traverse_reports_a_small_bufferfunction#

View source

def test_fmm_traverse_reports_a_small_buffer(tree)

test_soften_pairsfunction#

View source

def test_soften_pairs(system)

test_soften_pairs_reports_coincident_chargesfunction#

View source

def test_soften_pairs_reports_coincident_charges()

test_tree_build_kernels_and_momentsfunction#

View source

def test_tree_build_kernels_and_moments(system, tree)

treefunction#

View source

def tree(system)