Skip to content

Commit 13bcfab

Browse files
committed
pml/cm: Fix call convertor_prepare_for_send to use the right pointer
Previously did not use the right offset, same as 36a35fb Signed-off-by: William Zhang <wilzhang@amazon.com>
1 parent e40ca0d commit 13bcfab

File tree

1 file changed

+8
-16
lines changed

1 file changed

+8
-16
lines changed

ompi/mca/pml/cm/pml_cm_sendreq.h

Lines changed: 8 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -239,22 +239,14 @@ do { \
239239
ompi_mpi_local_convertor->flags; \
240240
(req_send)->req_base.req_convertor.master = \
241241
ompi_mpi_local_convertor->master; \
242-
(req_send)->req_base.req_convertor.local_size = \
243-
count * datatype->super.size; \
244-
(req_send)->req_base.req_convertor.pBaseBuf = \
245-
(unsigned char*)buf + datatype->super.true_lb; \
246-
(req_send)->req_base.req_convertor.count = count; \
247-
(req_send)->req_base.req_convertor.pDesc = &datatype->super; \
248-
if (opal_built_with_cuda_support) { \
249-
/* Switches off device buffer detection if \
250-
MTL set MCA_MTL_BASE_FLAG_ACCELERATOR_INIT_DISABLE during init */ \
251-
MCA_PML_CM_SWITCH_ACCELERATOR_CONVERTOR_OFF(flags, datatype, count); \
252-
(req_send)->req_base.req_convertor.flags |= flags; \
253-
/* Sets CONVERTOR_ACCELERATOR flag if device buffer */ \
254-
opal_convertor_prepare_for_send( \
255-
&req_send->req_base.req_convertor, \
256-
&datatype->super, count, buf ); \
257-
} \
242+
/* Switches off device buffer detection if \
243+
MTL set MCA_MTL_BASE_FLAG_ACCELERATOR_INIT_DISABLE during init */ \
244+
MCA_PML_CM_SWITCH_ACCELERATOR_CONVERTOR_OFF(flags, datatype, count); \
245+
(req_send)->req_base.req_convertor.flags |= flags; \
246+
/* Sets CONVERTOR_ACCELERATOR flag if device buffer */ \
247+
opal_convertor_prepare_for_send( \
248+
&req_send->req_base.req_convertor, \
249+
&datatype->super, count, (unsigned char*)buf + datatype->super.true_lb); \
258250
} else { \
259251
MCA_PML_CM_SWITCH_ACCELERATOR_CONVERTOR_OFF(flags, datatype, count); \
260252
opal_convertor_copy_and_prepare_for_send( \

0 commit comments

Comments
 (0)