|
20 | 20 | #include <ucs/sys/ptr_arith.h> |
21 | 21 | #include <uct/cuda/base/cuda_ctx.inl> |
22 | 22 | #include <uct/api/v2/uct_v2.h> |
23 | | -#include <uct/base/uct_iface.h> |
24 | 23 | #include <cuda.h> |
25 | 24 | #if CUDA_VERSION >= 11070 |
26 | 25 | #include <cudaTypedefs.h> |
@@ -119,14 +118,7 @@ uct_cuda_copy_md_query(uct_md_h uct_md, uct_md_attr_v2_t *md_attr) |
119 | 118 | uct_cuda_copy_md_t *md = ucs_derived_of(uct_md, uct_cuda_copy_md_t); |
120 | 119 |
|
121 | 120 | uct_md_base_md_query(md_attr); |
122 | | - /* Advertise RMA invalidation so RMA BW lanes qualify for UCP wireup when |
123 | | - * UCP_EP_INIT_ERR_MODE_PEER_FAILURE is set with RNDV (see |
124 | | - * ucp_wireup_add_rma_bw_lanes). cuda_copy has no remote RMA keys; local |
125 | | - * registration teardown is synchronous and satisfies the invalidate |
126 | | - * contract via uct_cuda_copy_mem_dereg(). */ |
127 | | - md_attr->flags = UCT_MD_FLAG_REG | UCT_MD_FLAG_ALLOC | |
128 | | - UCT_MD_FLAG_INVALIDATE | |
129 | | - UCT_MD_FLAG_INVALIDATE_RMA; |
| 121 | + md_attr->flags = UCT_MD_FLAG_REG | UCT_MD_FLAG_ALLOC; |
130 | 122 | md_attr->reg_mem_types = UCS_BIT(UCS_MEMORY_TYPE_HOST) | |
131 | 123 | UCS_BIT(UCS_MEMORY_TYPE_CUDA) | |
132 | 124 | UCS_BIT(UCS_MEMORY_TYPE_CUDA_MANAGED); |
@@ -188,21 +180,13 @@ UCS_PROFILE_FUNC(ucs_status_t, uct_cuda_copy_mem_dereg, |
188 | 180 | (md, params), |
189 | 181 | uct_md_h md, const uct_md_mem_dereg_params_t *params) |
190 | 182 | { |
191 | | - unsigned flags; |
192 | | - |
193 | | - UCT_MD_MEM_DEREG_CHECK_PARAMS(params, 1); |
| 183 | + UCT_MD_MEM_DEREG_CHECK_PARAMS(params, 0); |
194 | 184 |
|
195 | 185 | if (params->memh != &uct_cuda_dummy_memh) { |
196 | 186 | UCT_CUDADRV_FUNC(cuMemHostUnregister((void*)params->memh), |
197 | 187 | UCS_LOG_LEVEL_DIAG); |
198 | 188 | } |
199 | 189 |
|
200 | | - flags = UCT_MD_MEM_DEREG_FIELD_VALUE(params, flags, FIELD_FLAGS, 0); |
201 | | - if (flags & UCT_MD_MEM_DEREG_FLAG_INVALIDATE) { |
202 | | - ucs_assert(params->comp != NULL); /* suppress coverity false-positive */ |
203 | | - uct_invoke_completion(params->comp, UCS_OK); |
204 | | - } |
205 | | - |
206 | 190 | return UCS_OK; |
207 | 191 | } |
208 | 192 |
|
|
0 commit comments