From c90173bc105c11deb21d23f57f69065398d53efc Mon Sep 17 00:00:00 2001 From: Shi Jin Date: Wed, 8 Nov 2023 22:18:09 +0000 Subject: [PATCH] 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 (cherry picked from commit 1f40ede770e7a2dda621a335e9c2f558391f9b46) --- fabtests/pytest/efa/test_mr.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/fabtests/pytest/efa/test_mr.py b/fabtests/pytest/efa/test_mr.py index 5d03743f86a..f47c2d3fafe 100644 --- 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()