Check if receive buffer is cuda before staging at root

Akshay-Venkatesh · Akshay-Venkatesh · commit 8d9e60fdeeec · 2022-04-27T18:50:19.000-07:00
Signed-off-by: Akshay Venkatesh &lt;akshay.v.3.14@gmail.com&gt;
diff --git a/ompi/mca/coll/cuda/coll_cuda_reduce.c b/ompi/mca/coll/cuda/coll_cuda_reduce.c
@@ -55,7 +55,7 @@ mca_coll_cuda_reduce(const void *sbuf, void *rbuf, int count,
         sbuf = sbuf1 - gap;
     }
 
-    if (rank == root) {
+    if ((rank == root) && (opal_cuda_check_bufs((char *)rbuf, NULL))) {
         rbuf1 = (char*)malloc(bufsize);
         if (NULL == rbuf1) {
             if (NULL != sbuf1) free(sbuf1);

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@ mca_coll_cuda_reduce(const void sbuf, void rbuf, int count,`
`55`	`55`	`sbuf = sbuf1 - gap;`
`56`	`56`	`}`
`57`	`57`
`58`		`- if (rank == root) {`
	`58`	`+ if ((rank == root) && (opal_cuda_check_bufs((char *)rbuf, NULL))) {`
`59`	`59`	`rbuf1 = (char*)malloc(bufsize);`
`60`	`60`	`if (NULL == rbuf1) {`
`61`	`61`	`if (NULL != sbuf1) free(sbuf1);`