Skip to content

Commit

Permalink
TL/CUDA: fix CI build
Browse files Browse the repository at this point in the history
  • Loading branch information
ikryukov committed Nov 18, 2024
1 parent 4214598 commit 6bf9740
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
6 changes: 4 additions & 2 deletions src/components/tl/cuda/bcast/bcast_linear.c
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,8 @@ void ucc_tl_cuda_bcast_linear_progress(ucc_coll_task_t *coll_task)
int i;
ucc_rank_t peer;
ucc_tl_cuda_shm_barrier_t *curr_bar;
bool found;


task->super.status = UCC_INPROGRESS;

Expand All @@ -109,7 +111,7 @@ void ucc_tl_cuda_bcast_linear_progress(ucc_coll_task_t *coll_task)
switch (task->bcast_linear.stage) {
case STAGE_INIT_BAR_ROOT:
if (UCC_COLL_ARGS_ACTIVE_SET(&TASK_ARGS(task))) {
bool found = false;
found = false;
peer = ucc_ep_map_eval(task->subset.map, 1);
/* search first free barrier in active set pool */
for (i = 0; i < max_concurrent; ++i) {
Expand Down Expand Up @@ -140,7 +142,7 @@ void ucc_tl_cuda_bcast_linear_progress(ucc_coll_task_t *coll_task)
break; // TODO: move all logic to separate functions
}
case STAGE_FIND_BAR_PEER:
bool found = false;
found = false;
for (i = 0; i < max_concurrent; ++i) {
curr_bar = UCC_TL_CUDA_TEAM_BARRIER(team, max_concurrent + i);
if (curr_bar->tag == task->bcast_linear.key) {
Expand Down
1 change: 0 additions & 1 deletion src/components/tl/cuda/tl_cuda_coll.h
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,6 @@ static inline void ucc_tl_cuda_task_put(ucc_tl_cuda_task_t *task)
UCC_TL_CUDA_PROFILE_REQUEST_FREE(task);

if (UCC_TL_TEAM_RANK(TASK_TEAM(task)) == task->bcast_linear.root) {
ucc_print("free bar!");
task->bar->tag = UCC_TAG_FREE;
}

Expand Down

0 comments on commit 6bf9740

Please sign in to comment.