Skip to content

Commit

Permalink
Merge pull request #9374 from brminich/ucp/generic_cuda_rndv_fix_1.15.x
Browse files Browse the repository at this point in the history
UCP/RNDV: Do not use recv ppln with generic send buf - v1.15.x
  • Loading branch information
yosefe authored Sep 19, 2023
2 parents 9b3aeaa + 9d41061 commit d9891ae
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/ucp/rndv/rndv.c
Original file line number Diff line number Diff line change
Expand Up @@ -122,15 +122,15 @@ ucp_rndv_is_put_pipeline_needed(uintptr_t remote_address, size_t length,
const ucp_ep_rndv_zcopy_config_t *put_zcopy,
int is_get_zcopy_failed)
{
if ((remote_address != 0) &&
if ((remote_address == 0) ||
(ucp_rkey_packed_mem_type(rkey_buf) == UCS_MEMORY_TYPE_HOST)) {
return 0;
}

/* Fallback to PUT pipeline if: */
return /* Remote mem type is non-HOST memory OR can't do GET ZCOPY */
((remote_address == 0) || (get_zcopy->max == 0) ||
(length < get_zcopy->min) || is_get_zcopy_failed) &&
((get_zcopy->max == 0) || (length < get_zcopy->min) ||
is_get_zcopy_failed) &&
/* AND can do PUT assuming that configurations are symmetric */
((put_zcopy->max != 0) && (length >= put_zcopy->min));
}
Expand Down

0 comments on commit d9891ae

Please sign in to comment.