Skip to content

Commit 42d50e7

Browse files
committed
fix team problem
1 parent 8f0e1da commit 42d50e7

File tree

1 file changed

+4
-12
lines changed

1 file changed

+4
-12
lines changed

src/components/tl/ucp/allgather/allgather_knomial.c

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)