@@ -170,7 +170,7 @@ int mca_coll_ucx_allreduce(const void *sbuf, void *rbuf, int count,
170
170
ompi_datatype_type_extent (dtype , & extent );
171
171
ucs_status_t ret = mca_coll_ucx_check_total_data_size ((size_t )extent , count );
172
172
if (OPAL_UNLIKELY (ret != UCS_OK )) {
173
- COLL_UCX_WARN ( "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
173
+ COLL_UCX_VERBOSE ( 1 , "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
174
174
goto fallback ;
175
175
}
176
176
@@ -195,13 +195,13 @@ int mca_coll_ucx_allreduce(const void *sbuf, void *rbuf, int count,
195
195
ret = ucg_coll_allreduce_init (sbuf_rel , rbuf , count , (size_t )extent , dtype , ucx_module -> ucg_group , 0 ,
196
196
op , 0 , 0 , & coll );
197
197
if (OPAL_UNLIKELY (ret != UCS_OK )) {
198
- COLL_UCX_WARN ( "ucx allreduce init failed: %s, fallback to ompi function." , ucs_status_string (ret ));
198
+ COLL_UCX_VERBOSE ( 1 , "ucx allreduce init failed: %s, fallback to ompi function." , ucs_status_string (ret ));
199
199
goto fallback ;
200
200
}
201
201
202
202
ret = ucg_collective_start_nbr (coll , req );
203
203
if (OPAL_UNLIKELY (UCS_STATUS_IS_ERR (ret ))) {
204
- COLL_UCX_WARN ( "ucx allreduce start failed: %s, fallback to ompi function." , ucs_status_string (ret ));
204
+ COLL_UCX_VERBOSE ( 1 , "ucx allreduce start failed: %s, fallback to ompi function." , ucs_status_string (ret ));
205
205
goto fallback ;
206
206
}
207
207
@@ -550,13 +550,13 @@ int mca_coll_ucx_alltoallv(const void *sbuf, const int *scounts, const int *sdis
550
550
is_recv_contig = mca_coll_ucg_check_contig_datatype (rdtype );
551
551
552
552
if (!is_send_contig || !is_recv_contig ) {
553
- COLL_UCX_WARN ( "current hmpi alltoallv cannot support non-contig datatype, fallback to ompi function." );
553
+ COLL_UCX_VERBOSE ( 1 , "current hmpi alltoallv cannot support non-contig datatype, fallback to ompi function." );
554
554
goto fallback ;
555
555
}
556
556
557
557
/* large datatype */
558
558
if (sdtype_size > LARGE_DATATYPE_THRESHOLD || rdtype_size > LARGE_DATATYPE_THRESHOLD ) {
559
- COLL_UCX_WARN ( "current hmpi alltoallv cannot support large datatype, fallback to ompi function." );
559
+ COLL_UCX_VERBOSE ( 1 , "current hmpi alltoallv cannot support large datatype, fallback to ompi function." );
560
560
goto fallback ;
561
561
}
562
562
@@ -570,7 +570,7 @@ int mca_coll_ucx_alltoallv(const void *sbuf, const int *scounts, const int *sdis
570
570
}
571
571
ret = mca_coll_ucx_check_total_data_size ((size_t )sdtype_size , total_send_count );
572
572
if (OPAL_UNLIKELY (ret != UCS_OK )) {
573
- COLL_UCX_WARN ( "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
573
+ COLL_UCX_VERBOSE ( 1 , "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
574
574
goto fallback ;
575
575
}
576
576
@@ -580,7 +580,7 @@ int mca_coll_ucx_alltoallv(const void *sbuf, const int *scounts, const int *sdis
580
580
}
581
581
ret = mca_coll_ucx_check_total_data_size ((size_t )rdtype_size , total_recv_count );
582
582
if (OPAL_UNLIKELY (ret != UCS_OK )) {
583
- COLL_UCX_WARN ( "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
583
+ COLL_UCX_VERBOSE ( 1 , "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
584
584
goto fallback ;
585
585
}
586
586
@@ -589,7 +589,7 @@ int mca_coll_ucx_alltoallv(const void *sbuf, const int *scounts, const int *sdis
589
589
rbuf , rcounts , (size_t )rdtype_size , rdtype , rdispls ,
590
590
ucx_module -> ucg_group , 0 , 0 , 0 , 0 , & coll );
591
591
if (OPAL_UNLIKELY (ret != UCS_OK )) {
592
- COLL_UCX_WARN ( "ucx alltoallv init failed: %s" , ucs_status_string (ret ));
592
+ COLL_UCX_VERBOSE ( 1 , "ucx alltoallv init failed: %s" , ucs_status_string (ret ));
593
593
goto fallback ;
594
594
}
595
595
@@ -601,7 +601,7 @@ int mca_coll_ucx_alltoallv(const void *sbuf, const int *scounts, const int *sdis
601
601
ucs_status_ptr_t req = request + mca_coll_ucx_component .request_size ;
602
602
ret = ucg_collective_start_nbr (coll , req );
603
603
if (OPAL_UNLIKELY (UCS_STATUS_IS_ERR (ret ))) {
604
- COLL_UCX_WARN ( "ucx alltoallv start failed: %s" , ucs_status_string (ret ));
604
+ COLL_UCX_VERBOSE ( 1 , "ucx alltoallv start failed: %s" , ucs_status_string (ret ));
605
605
goto fallback ;
606
606
}
607
607
@@ -635,13 +635,13 @@ int mca_coll_ucx_barrier(struct ompi_communicator_t *comm, mca_coll_base_module_
635
635
ucg_coll_h coll = NULL ;
636
636
ucs_status_t ret = ucg_coll_barrier_init (0 , ucx_module -> ucg_group , 0 , 0 , 0 , 0 , & coll );
637
637
if (OPAL_UNLIKELY (ret != UCS_OK )) {
638
- COLL_UCX_WARN ( "ucx barrier init failed: %s, fallback to ompi function." , ucs_status_string (ret ));
638
+ COLL_UCX_VERBOSE ( 1 , "ucx barrier init failed: %s, fallback to ompi function." , ucs_status_string (ret ));
639
639
goto fallback ;
640
640
}
641
641
642
642
ret = ucg_collective_start_nbr (coll , req );
643
643
if (OPAL_UNLIKELY (UCS_STATUS_IS_ERR (ret ))) {
644
- COLL_UCX_WARN ( "ucx barrier start failed: %s, fallback to ompi function." , ucs_status_string (ret ));
644
+ COLL_UCX_VERBOSE ( 1 , "ucx barrier start failed: %s, fallback to ompi function." , ucs_status_string (ret ));
645
645
goto fallback ;
646
646
}
647
647
@@ -677,19 +677,19 @@ int mca_coll_ucx_bcast(void *buff, int count, struct ompi_datatype_t *dtype, int
677
677
ompi_datatype_type_extent (dtype , & dtype_size );
678
678
ucs_status_t ret = mca_coll_ucx_check_total_data_size ((size_t )dtype_size , count );
679
679
if (OPAL_UNLIKELY (ret != UCS_OK )) {
680
- COLL_UCX_WARN ( "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
680
+ COLL_UCX_VERBOSE ( 1 , "ucx component only support data size <= 2^32 bytes, fallback to ompi function." );
681
681
goto fallback ;
682
682
}
683
683
ret = ucg_coll_bcast_init (buff , buff , count , (size_t )dtype_size , dtype , ucx_module -> ucg_group , 0 ,
684
684
0 , root , 0 , & coll );
685
685
if (OPAL_UNLIKELY (UCS_STATUS_IS_ERR (ret ))) {
686
- COLL_UCX_WARN ( "ucx bcast init failed: %s, fallback to ompi function." , ucs_status_string (ret ));
686
+ COLL_UCX_VERBOSE ( 1 , "ucx bcast init failed: %s, fallback to ompi function." , ucs_status_string (ret ));
687
687
goto fallback ;
688
688
}
689
689
690
690
ret = ucg_collective_start_nbr (coll , req );
691
691
if (OPAL_UNLIKELY (UCS_STATUS_IS_ERR (ret ))) {
692
- COLL_UCX_WARN ( "ucx bcast start failed: %s, fallback to ompi function." , ucs_status_string (ret ));
692
+ COLL_UCX_VERBOSE ( 1 , "ucx bcast start failed: %s, fallback to ompi function." , ucs_status_string (ret ));
693
693
goto fallback ;
694
694
}
695
695
0 commit comments