UCP: Fix send perf estimation

openucx · Sep 19, 2024 · 8ce6409 · 8ce6409
1 parent c1f718c
commit 8ce6409
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ucp/proto/proto_init.c b/src/ucp/proto/proto_init.c
@@ -497,8 +497,8 @@ ucp_proto_common_init_send_perf(const ucp_proto_common_init_params_t *params,
  ucs_assert(reg_md_map == 0);
 
  if ((params->flags & UCP_PROTO_COMMON_INIT_FLAG_REMOTE_ACCESS) &&
- (params->super.rkey_config_key != NULL)) {
- remote_mem_type = params->super.rkey_config_key->mem_type;
+ (params->reg_mem_type != UCS_MEMORY_TYPE_UNKNOWN)) {
+ remote_mem_type = params->reg_mem_type;
  } else {
  remote_mem_type = UCS_MEMORY_TYPE_HOST;
  }