Skip to content

Commit 102872c

Browse files
committed
fabtests/pytest/efa: Restrict the hmem iface for host only test
For host only test, we do not need to initialize non-system hmem ifaces, this will speed up the test setup on accelerator based platforms Signed-off-by: Shi Jin <sjina@amazon.com>
1 parent c7956e4 commit 102872c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

fabtests/pytest/efa/efa_common.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,10 @@ def efa_run_client_server_test(cmdline_args, executable, iteration_type,
175175
if "cuda" in memory_type:
176176
timeout = max(1000, timeout)
177177

178+
# For host only test, we do not need to initialize non-system hmem ifaces
179+
if memory_type == "host_to_host":
180+
additional_env += ' FI_HMEM=system'
181+
178182
test = ClientServerTest(cmdline_args, executable, iteration_type,
179183
completion_semantic=completion_semantic,
180184
datacheck_type="with_datacheck",

0 commit comments

Comments
 (0)