fabtests: Add pytests for FI_MORE

Test fi_rma_bw and fi_rdm_tagged_bw with flag FI_MORE.

Signed-off-by: Jessie Yang <[email protected]>

fabtests/pytest/efa/test_rdm.py

@@ -71,6 +71,11 @@ def test_rdm_tagged_bw_small_tx_rx(cmdline_args, completion_semantic, memory_typ
     efa_run_client_server_test(cmdline_args_copy, "fi_rdm_tagged_bw -W 128", "short",
                                completion_semantic, memory_type, "all", completion_type=completion_type)
 
+@pytest.mark.functional
+def test_rdm_tagged_bw_use_fi_more(cmdline_args, completion_semantic, memory_type, message_size):
+    efa_run_client_server_test(cmdline_args, "fi_rdm_tagged_bw --use-fi-more",
+                               "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)])

fabtests/pytest/efa/test_rma_bw.py

@@ -62,3 +62,13 @@ def test_rma_bw_1G(cmdline_args, operation_type, completion_semantic):
     efa_run_client_server_test(cmdline_args, command, 2,
                                completion_semantic=completion_semantic, message_size=1073741824,
                                memory_type="host_to_host", warmup_iteration_type=0, timeout=timeout)
+
+@pytest.mark.functional
+@pytest.mark.parametrize("operation_type", ["writedata", "write"])
+def test_rma_bw_use_fi_more(cmdline_args, operation_type, completion_semantic, inject_message_size):
+    command = "fi_rma_bw -e rdm -j 0 --use-fi-more"
+    command = command + " -o " + operation_type
+    # rma_bw test with data verification takes longer to finish
+    timeout = max(540, cmdline_args.timeout)
+    efa_run_client_server_test(cmdline_args, command, "short", completion_semantic,
+                               "host_to_host", inject_message_size, timeout=timeout)