1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
from default.test_rdm import test_rdm, test_rdm_bw_functional
from efa.efa_common import efa_run_client_server_test
import pytest
@pytest.mark.parametrize("iteration_type",
[pytest.param("short", marks=pytest.mark.short),
pytest.param("standard", marks=pytest.mark.standard)])
def test_rdm_pingpong(cmdline_args, iteration_type, completion_semantic, memory_type, completion_type):
efa_run_client_server_test(cmdline_args, "fi_rdm_pingpong", iteration_type,
completion_semantic, memory_type, "all", completion_type=completion_type)
@pytest.mark.functional
@pytest.mark.serial
def test_mr_exhaustion_rdm_pingpong(cmdline_args):
efa_run_client_server_test(cmdline_args, "fi_efa_exhaust_mr_reg_rdm_pingpong", "short",
"transmit_complete", "host_to_host", "all", timeout=1000)
@pytest.mark.functional
def test_rdm_pingpong_range(cmdline_args, completion_semantic, memory_type, message_size):
efa_run_client_server_test(cmdline_args, "fi_rdm_pingpong", "short",
completion_semantic, memory_type, message_size)
@pytest.mark.functional
def test_rdm_pingpong_no_inject_range(cmdline_args, completion_semantic, inject_message_size):
efa_run_client_server_test(cmdline_args, "fi_rdm_pingpong -j 0", "short",
completion_semantic, "host_to_host", inject_message_size)
@pytest.mark.parametrize("iteration_type",
[pytest.param("short", marks=pytest.mark.short),
pytest.param("standard", marks=pytest.mark.standard)])
def test_rdm_tagged_pingpong(cmdline_args, iteration_type, completion_semantic, memory_type, completion_type):
efa_run_client_server_test(cmdline_args, "fi_rdm_tagged_pingpong", iteration_type,
completion_semantic, memory_type, "all", completion_type=completion_type)
@pytest.mark.functional
def test_rdm_tagged_pingpong_range(cmdline_args, completion_semantic, memory_type, message_size):
efa_run_client_server_test(cmdline_args, "fi_rdm_tagged_pingpong", "short",
completion_semantic, memory_type, message_size)
@pytest.mark.parametrize("iteration_type",
[pytest.param("short", marks=pytest.mark.short),
pytest.param("standard", marks=pytest.mark.standard)])
def test_rdm_tagged_bw(cmdline_args, iteration_type, completion_semantic, memory_type, completion_type):
efa_run_client_server_test(cmdline_args, "fi_rdm_tagged_bw", iteration_type,
completion_semantic, memory_type, "all", completion_type=completion_type)
@pytest.mark.functional
def test_rdm_tagged_bw_range(cmdline_args, completion_semantic, memory_type, message_size):
efa_run_client_server_test(cmdline_args, "fi_rdm_tagged_bw", "short",
completion_semantic, memory_type, message_size)
@pytest.mark.functional
def test_rdm_tagged_bw_no_inject_range(cmdline_args, completion_semantic, inject_message_size):
efa_run_client_server_test(cmdline_args, "fi_rdm_tagged_bw -j 0", "short",
completion_semantic, "host_to_host", inject_message_size)
@pytest.mark.parametrize("iteration_type",
[pytest.param("short", marks=pytest.mark.short),
pytest.param("standard", marks=pytest.mark.standard)])
def test_rdm_atomic(cmdline_args, iteration_type, completion_semantic, memory_type):
from copy import copy
from common import ClientServerTest
if "neuron" in memory_type:
pytest.skip("Neuron does not fully support atomics")
# the rdm_atomic test's run time has a high variance when running single c6gn instance.
# the issue is tracked in: https://github.com/ofiwg/libfabric/issues/7002
# to mitigate the issue, set the maximum timeout of fi_rdm_atomic to 1800 seconds.
cmdline_args_copy = copy(cmdline_args)
test = ClientServerTest(cmdline_args_copy, "fi_rdm_atomic", iteration_type, completion_semantic,
memory_type=memory_type, timeout=1800)
test.run()
@pytest.mark.functional
def test_rdm_tagged_peek(cmdline_args):
from copy import copy
from common import ClientServerTest
test = ClientServerTest(cmdline_args, "fi_rdm_tagged_peek", timeout=1800)
test.run()
|
