Skip to content

Commit

Permalink
fabtests/pytest/efa: Skip memory registration that hit device limit
Browse files Browse the repository at this point in the history
The EFA limit for single MR that enables remote write is 1M pages aka 4GB for regular pages.
Currently test_unexpected_msg can hit this limit for certain msg_size & msg_count combo.
This patch skips such combos that can hit the limit to avoid errors.

Signed-off-by: Shi Jin <sjina@amazon.com>
  • Loading branch information
shijin-aws committed Sep 19, 2024
1 parent 677bdeb commit ade8d60
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions fabtests/pytest/efa/test_unexpected_msg.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,12 @@ def test_unexpected_msg(cmdline_args, msg_size, msg_count, memory_type, completi
neuron_maximal_buffer_size = 2**32
if "neuron" in memory_type and allocated_memory >= neuron_maximal_buffer_size:
pytest.skip("Cannot hit neuron allocation limit")

# The EFA limit for single MR that enables remote write is 1M pages aka 4GB for regular pages
maximal_mr_size = 2**32
if allocated_memory >= maximal_mr_size:
pytest.skip("Cannot hit EFA MR limit")

efa_run_client_server_test(cmdline_args, f"fi_unexpected_msg -e rdm -M {msg_count}", iteration_type="short",
completion_semantic=completion_semantic, memory_type=memory_type,
message_size=msg_size, completion_type="queue", timeout=1800)

0 comments on commit ade8d60

Please sign in to comment.