fabtests/pytests/efa: onboard dmabuf argument for test_mr

make test_mr recognize the --do-dmabuf-reg-for-hmem cmdline argument. Signed-off-by: Shi Jin <[email protected]> (cherry picked from commit 1f40ede)
shijin-aws · Nov 9, 2023 · c90173b · c90173b
1 parent 756fa74
commit c90173b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/fabtests/pytest/efa/test_mr.py b/fabtests/pytest/efa/test_mr.py
@@ -26,9 +26,14 @@ def test_mr_hmem(cmdline_args, hmem_type):
 
     cmdline_args_copy = copy.copy(cmdline_args)
 
+    test_command = f"fi_mr_test -D {hmem_type}"
+
+    if cmdline_args.do_dmabuf_reg_for_hmem:
+        test_command += " -R"
+
     test = UnitTest(
         cmdline_args_copy,
-        f"fi_mr_test -D {hmem_type}",
+        test_command,
         failing_warn_msgs=["Unable to add MR to map"],
     )
     test.run()