fabtests/pytest: use different message sizes for rma pingpong

Signed-off-by: Jessie Yang <jiaxiyan@amazon.com>
jiaxiyan · Jun 6, 2024 · 2a37e0c · 2a37e0c
1 parent 166c2b6
commit 2a37e0c
Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/fabtests/pytest/efa/test_rma_pingpong.py b/fabtests/pytest/efa/test_rma_pingpong.py
@@ -19,6 +19,9 @@ def test_rma_pingpong(cmdline_args, iteration_type, operation_type, completion_s
 
 @pytest.mark.functional
 @pytest.mark.parametrize("operation_type", ["writedata", "write"])
+@pytest.mark.parametrize("message_size", ["r:4048,4,4148",
+                                          "r:8000,4,9000",
+                                          "r:17000,4,18000"])
 def test_rma_pingpong_range(cmdline_args, operation_type, completion_semantic, message_size, memory_type):
     if memory_type != "host_to_host" and operation_type == "write":
         pytest.skip("no hmem memory support for pingpong_rma write test")
@@ -31,6 +34,9 @@ def test_rma_pingpong_range(cmdline_args, operation_type, completion_semantic, m
 
 @pytest.mark.functional
 @pytest.mark.parametrize("operation_type", ["writedata", "write"])
+@pytest.mark.parametrize("message_size", ["r:4048,4,4148",
+                                          "r:8000,4,9000",
+                                          "r:17000,4,18000"])
 def test_rma_pingpong_range_no_inject(cmdline_args, operation_type, completion_semantic, inject_message_size, memory_type):
     if memory_type != "host_to_host" and operation_type == "write":
         pytest.skip("no hmem memory support for pingpong_rma write test")

diff --git a/fabtests/pytest/shm/test_rma_pingpong.py b/fabtests/pytest/shm/test_rma_pingpong.py
@@ -18,6 +18,9 @@ def test_rma_pingpong(cmdline_args, iteration_type, operation_type, completion_s
 
 @pytest.mark.functional
 @pytest.mark.parametrize("operation_type", ["writedata", "write"])
+@pytest.mark.parametrize("message_size", ["r:4048,4,4148",
+                                          "r:8000,4,9000",
+                                          "r:17000,4,18000"])
 def test_rma_pingpong_range(cmdline_args, operation_type, completion_semantic, message_size, memory_type):
     if memory_type != "host_to_host" and operation_type == "write":
         pytest.skip("no hmem memory support for pingpong_rma write test")