@@ -60,14 +60,6 @@ void ucc_tl_ucp_allgather_knomial_progress(ucc_coll_task_t *coll_task)
6060 ucc_status_t status ;
6161 size_t extra_count ;
6262
63- uint32_t USE_CUDA = UCC_TL_UCP_TEAM_LIB (team )-> cfg .allgather_use_cuda ;
64- if (!USE_CUDA ){
65- if (UCC_INPROGRESS == ucc_tl_ucp_test (task )){
66- // should I use ucc_tl_ucp_test_with_etasks ?
67- return ;
68- }
69- }
70-
7163 EXEC_TASK_TEST (UCC_KN_PHASE_INIT , "failed during ee task test" ,
7264 task -> allgather_kn .etask );
7365 task -> allgather_kn .etask = NULL ;
@@ -418,12 +410,12 @@ ucc_status_t ucc_tl_ucp_allgather_knomial_init_r(
418410 ucc_base_coll_args_t * coll_args , ucc_base_team_t * team ,
419411 ucc_coll_task_t * * task_h , ucc_kn_radix_t radix )
420412{
421- ucc_tl_ucp_team_t * tl_team = ucc_derived_of (team , ucc_tl_ucp_team_t );
422- ucc_tl_ucp_task_t * task ;
413+ ucc_tl_ucp_team_t * tl_team = ucc_derived_of (team , ucc_tl_ucp_team_t );
414+ ucc_tl_ucp_task_t * task = ucc_tl_ucp_init_task ( coll_args , team ) ;
423415 ucc_sbgp_t * sbgp ;
424416 ucc_status_t status ;
425- int use_loopback = UCC_TL_UCP_TEAM_LIB ( team ) -> cfg . allgather_use_loopback ;
426- task = ucc_tl_ucp_init_task ( coll_args , team ) ;
417+ ucc_tl_ucp_team_t * team_loopback = TASK_TEAM ( task ) ;
418+ int use_loopback = UCC_TL_UCP_TEAM_LIB ( team_loopback ) -> cfg . allgather_use_loopback ;
427419 status = ucc_mpool_init (& task -> allgather_kn .etask_node_mpool , 0 , sizeof (node_ucc_ee_executor_task_t ),
428420 0 , UCC_CACHE_LINE_SIZE , 16 , UINT_MAX , NULL ,
429421 tl_team -> super .super .context -> ucc_context -> thread_mode , "etasks_linked_list_nodes" );
0 commit comments