TL/CUDA: removed done stage

openucx · Aug 23, 2024 · 11c7311 · 11c7311
1 parent baa3f6e
commit 11c7311
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/src/components/tl/cuda/bcast/bcast_linear.c b/src/components/tl/cuda/bcast/bcast_linear.c
@@ -7,7 +7,6 @@
 #include "bcast/bcast.h"
 
 enum {
- STAGE_DONE,
  STAGE_SYNC,
  STAGE_SETUP,
  // root
@@ -178,11 +177,9 @@ void ucc_tl_cuda_bcast_linear_progress(ucc_coll_task_t *coll_task)
  return;
  } else {
  // finish
- task->bcast_linear.stage = STAGE_DONE;
+ task->super.status = UCC_OK;
+ break;
  }
- case STAGE_DONE:
- task->super.status = UCC_OK;
- break;
  default:
  break;
  }
@@ -225,17 +222,16 @@ void ucc_tl_cuda_bcast_linear_progress(ucc_coll_task_t *coll_task)
  task->bcast_linear.stage = STAGE_WAIT_ROOT;
  return;
  } else {
- task->bcast_linear.stage = STAGE_DONE;
+ // Done
+ task->super.status = UCC_OK;
+ break;
  }
  } else {
  return;
  }
  } else {
  return;
  }
- case STAGE_DONE:
- task->super.status = UCC_OK;
- break;
  default:
  break;
  }