fabtests: use different message sizes for rma pingpong

Signed-off-by: Jessie Yang <[email protected]>
jiaxiyan · Jun 6, 2024 · fce1790 · fce1790
1 parent 166c2b6
commit fce1790
Show file tree

Hide file tree

Showing 2 changed files with 10 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,11 @@ 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:1,4,65",
+                                          "r:4048,4,4148",
+                                          "r:8000,4,9000",
+                                          "r:17000,4,18000",
+                                          "r:1,1024,1048577"])
 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")

diff --git a/fabtests/pytest/shm/test_rma_pingpong.py b/fabtests/pytest/shm/test_rma_pingpong.py
@@ -18,6 +18,11 @@ 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:1,4,65",
+                                          "r:4048,4,4148",
+                                          "r:8000,4,9000",
+                                          "r:17000,4,18000",
+                                          "r:1,1024,1048577"])
 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")