v5.0.x: coll/cuda Fix the size_t/int parameter

janjust · janjust · commit 399f69d68735 · 2024-11-25T12:40:23.000-07:00
Signed-off-by: Tomislav Janjusic &lt;tomislavj@nvidia.com&gt;
diff --git a/ompi/mca/coll/cuda/coll_cuda.h b/ompi/mca/coll/cuda/coll_cuda.h
@@ -54,7 +54,7 @@ int mca_coll_cuda_reduce(const void *sbuf, void *rbuf, int count,
                          struct ompi_communicator_t *comm,
                          mca_coll_base_module_t *module);
 
-int mca_coll_cuda_reduce_local(const void *sbuf, void *rbuf, size_t count,
+int mca_coll_cuda_reduce_local(const void *sbuf, void *rbuf, int count,
                                struct ompi_datatype_t *dtype,
                                struct ompi_op_t *op,
                                mca_coll_base_module_t *module);
diff --git a/ompi/mca/coll/cuda/coll_cuda_reduce.c b/ompi/mca/coll/cuda/coll_cuda_reduce.c
@@ -85,7 +85,7 @@ mca_coll_cuda_reduce(const void *sbuf, void *rbuf, int count,
 }
 
 int
-mca_coll_cuda_reduce_local(const void *sbuf, void *rbuf, size_t count,
+mca_coll_cuda_reduce_local(const void *sbuf, void *rbuf, int count,
                            struct ompi_datatype_t *dtype,
                            struct ompi_op_t *op,
                            mca_coll_base_module_t *module)

Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ mca_coll_cuda_reduce(const void sbuf, void rbuf, int count,`
`85`	`85`	`}`
`86`	`86`
`87`	`87`	`int`
`88`		`-mca_coll_cuda_reduce_local(const void sbuf, void rbuf, size_t count,`
	`88`	`+mca_coll_cuda_reduce_local(const void sbuf, void rbuf, int count,`
`89`	`89`	`struct ompi_datatype_t *dtype,`
`90`	`90`	`struct ompi_op_t *op,`
`91`	`91`	`mca_coll_base_module_t *module)`