diff --git a/projects/rccl/src/collectives/device/all_reduce.h b/projects/rccl/src/collectives/device/all_reduce.h index f71953aa26..465fe32f54 100644 --- a/projects/rccl/src/collectives/device/all_reduce.h +++ b/projects/rccl/src/collectives/device/all_reduce.h @@ -691,8 +691,7 @@ struct RunWorkElement struct RunWorkElement { __device__ __forceinline__ void run(ncclWorkElem *args) { - if (args->pad_0 == 0) runTreeUpDown(args); - else runTreeSplit(args); + runTreeSplit(args); } };