OSC/UCX: Fix data validation issue with derived dt in get accumulate

Mamzi Bayatpour  mbayatpour@nvidia.com () · janjust · Mamzi Bayatpour  mbayatpour@nvidia.com () · commit 0031008f3ccd · 2022-05-19T14:04:26.000-07:00
Signed-off-by: Mamzi Bayatpour  &lt;mbayatpour@nvidia.com&gt;
Co-authored-by: Tomislav Janjusic &lt;tomislavj@nvidia.com&gt;
diff --git a/ompi/mca/osc/ucx/osc_ucx_comm.c b/ompi/mca/osc/ucx/osc_ucx_comm.c
@@ -990,6 +990,7 @@ int get_accumulate_req(const void *origin_addr, int origin_count,
                 if (ret != OMPI_SUCCESS) {
                     return ret;
                 }
+                temp_count *= target_count;
             }
             ompi_datatype_get_true_extent(temp_dt, &temp_lb, &temp_extent);
             temp_addr = free_addr = malloc(temp_extent * temp_count);

Original file line number	Diff line number	Diff line change
`@@ -990,6 +990,7 @@ int get_accumulate_req(const void *origin_addr, int origin_count,`
`990`	`990`	`if (ret != OMPI_SUCCESS) {`
`991`	`991`	`return ret;`
`992`	`992`	`}`
	`993`	`+ temp_count *= target_count;`
`993`	`994`	`}`
`994`	`995`	`ompi_datatype_get_true_extent(temp_dt, &temp_lb, &temp_extent);`
`995`	`996`	`temp_addr = free_addr = malloc(temp_extent * temp_count);`