Only use split tree algorithm to reduce kernel code size.

[ROCm/rccl commit: 02929cffb6]
이 커밋은 다음에 포함됨:
Wenkai Du 작성하고 Wen-Heng (Jack) Chung가 2022-09-22 12:01:53 -05:00를 커밋함
1 부모 e4d46a0f64
커밋 abdc365a05
1 파일이 변경됨
+1 -2
+1 -2
파일 보기
@@ -691,8 +691,7 @@ struct RunWorkElement<ncclFuncAllReduce, T, RedOp, NCCL_ALGO_RING, NCCL_PROTO_LL
template<typename T, typename RedOp>
struct RunWorkElement<ncclFuncAllReduce, T, RedOp, NCCL_ALGO_TREE, NCCL_PROTO_LL> {
__device__ __forceinline__ void run(ncclWorkElem *args) {
if (args->pad_0 == 0) runTreeUpDown<T, RedOp, ProtoLL>(args);
else runTreeSplit<T, RedOp, ProtoLL>(args);
runTreeSplit<T, RedOp, ProtoLL>(args);
}
};