Skip to content

Commit

Permalink
Merge pull request #9266 from yosefe/topic/tools-perf-fix-perftest-me…
Browse files Browse the repository at this point in the history
…mory-access-flags-v1.15.x

TOOLS/PERF: Fix perftest memory access flags - v1.15.x
  • Loading branch information
yosefe authored Aug 4, 2023
2 parents 7053c5e + 3fcdc27 commit 764e4e9
Showing 1 changed file with 22 additions and 3 deletions.
25 changes: 22 additions & 3 deletions src/tools/perf/lib/libperf_memory.c
Original file line number Diff line number Diff line change
Expand Up @@ -221,9 +221,28 @@ ucs_status_t uct_perf_test_alloc_mem(ucx_perf_context_t *perf)

/* TODO use params->alignment */

flags = (params->flags & UCX_PERF_TEST_FLAG_MAP_NONBLOCK) ?
UCT_MD_MEM_FLAG_NONBLOCK : 0;
flags |= UCT_MD_MEM_ACCESS_RMA;
flags = UCT_MD_MEM_ACCESS_LOCAL_READ | UCT_MD_MEM_ACCESS_LOCAL_WRITE;

switch (perf->params.command) {
case UCX_PERF_CMD_PUT:
flags |= UCT_MD_MEM_ACCESS_REMOTE_PUT;
break;
case UCX_PERF_CMD_GET:
flags |= UCT_MD_MEM_ACCESS_REMOTE_GET;
break;
case UCX_PERF_CMD_ADD:
case UCX_PERF_CMD_FADD:
case UCX_PERF_CMD_SWAP:
case UCX_PERF_CMD_CSWAP:
flags |= UCT_MD_MEM_ACCESS_REMOTE_ATOMIC;
break;
default:
break;
}

if (params->flags & UCX_PERF_TEST_FLAG_MAP_NONBLOCK) {
flags |= UCT_MD_MEM_FLAG_NONBLOCK;
}

/* Allocate send buffer memory */
status = perf->send_allocator->uct_alloc(perf,
Expand Down

0 comments on commit 764e4e9

Please sign in to comment.