TL/CUDA: fix triggered post a2a

Author: ikryukov

src/components/tl/cuda/alltoall/alltoall_ce.c

@@ -38,8 +38,6 @@ ucc_status_t ucc_tl_cuda_alltoall_ce_init(ucc_tl_cuda_task_t *task)
     size_t              data_len;
     int                 i;
 
-    task->super.flags |= UCC_COLL_TASK_FLAG_EXECUTOR;
-
     task->alltoallv_ce.get_size   = ucc_tl_cuda_alltoall_get_size;
     task->alltoallv_ce.get_offset = ucc_tl_cuda_alltoall_get_offset;
     task->alltoallv_ce.sdt        = args->src.info.datatype;
@@ -70,6 +68,8 @@ ucc_status_t ucc_tl_cuda_alltoall_ce_init(ucc_tl_cuda_task_t *task)
 
     if (lib->cfg.alltoall_use_copy_engine) {
         ucc_debug("ucc_tl_cuda_alltoallv_ce_init: copy engine");
+        task->super.triggered_post = ucc_tl_cuda_alltoallv_ce_triggered_post;
+
         task->alltoallv_ce.copy_post = cuda_copy_post;
         task->alltoallv_ce.evtCompletions = (cudaEvent_t*)ucc_malloc(team->num_streams * sizeof(cudaEvent_t), "alltoallv_ce.evtCompletions");
         for (i = 0; i < team->num_streams; i++) {

src/components/tl/cuda/alltoallv/alltoallv.h

@@ -62,4 +62,6 @@ ucc_status_t ee_copy_post(void *dst, void *src, size_t len,
                        ucc_ee_executor_t       *executor,
                        ucc_ee_executor_task_t **task, cudaStream_t stream);
 
+ucc_status_t ucc_tl_cuda_alltoallv_ce_triggered_post(ucc_ee_h ee, ucc_ev_t *ev,
+                                                     ucc_coll_task_t *coll_task);
 #endif

src/components/tl/cuda/alltoallv/alltoallv_ce.c

@@ -562,7 +562,7 @@ size_t ucc_tl_cuda_alltoallv_get_offset(const ucc_tl_cuda_task_t *task,
 
 //NOLINTNEXTLINE(misc-unused-parameters): ev parameter unused as it's not needed for this implementation
 ucc_status_t ucc_tl_cuda_alltoallv_ce_triggered_post(ucc_ee_h ee, ucc_ev_t *ev,
-                                        ucc_coll_task_t *coll_task)
+                                                     ucc_coll_task_t *coll_task)
 {
     ucc_tl_cuda_task_t *task  = ucc_derived_of(coll_task, ucc_tl_cuda_task_t);
     ucc_status_t        status;