Skip to content

Commit

Permalink
Merge pull request #10182 from ivankochin/tools/perf/use-mkey-pack-v2
Browse files Browse the repository at this point in the history
TOOLS/PERF: Use mkey_pack_v2
  • Loading branch information
yosefe authored Sep 28, 2024
2 parents 4234ca0 + 541bf6c commit 7d0bf01
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion src/tools/perf/lib/libperf.c
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
#include <ucs/sys/string.h>
#include <ucs/type/serialize.h>
#include <tools/perf/lib/libperf_int.h>
#include <uct/api/v2/uct_v2.h>

#include <string.h>
#include <unistd.h>
Expand Down Expand Up @@ -633,6 +634,9 @@ static ucs_status_t uct_perf_test_check_capabilities(ucx_perf_params_t *params,

static ucs_status_t uct_perf_test_setup_endpoints(ucx_perf_context_t *perf)
{
uct_md_mkey_pack_params_t mkey_params = {
.field_mask = 0
};
const size_t buffer_size = ADDR_BUF_SIZE;
ucx_perf_ep_info_t info, *remote_info;
unsigned group_size, i, group_index;
Expand Down Expand Up @@ -700,7 +704,10 @@ static ucs_status_t uct_perf_test_setup_endpoints(ucx_perf_context_t *perf)

if (md_attr.cap.flags & (UCT_MD_FLAG_ALLOC|UCT_MD_FLAG_REG)) {
memset(rkey_buffer, 0, info.rkey_size);
status = uct_md_mkey_pack(perf->uct.md, perf->uct.recv_mem.memh, rkey_buffer);
status = uct_md_mkey_pack_v2(perf->uct.md, perf->uct.recv_mem.memh,
perf->uct.recv_mem.address,
perf->uct.recv_mem.length, &mkey_params,
rkey_buffer);
if (status != UCS_OK) {
ucs_error("Failed to uct_rkey_pack: %s", ucs_status_string(status));
goto err_free;
Expand Down

0 comments on commit 7d0bf01

Please sign in to comment.