Skip to content

Commit 42d5dd5

Browse files
authored
[PAL] Add AVG reduction support to the Kunlunxin CCLAdaptor (flagos-ai#511)
1 parent fed5532 commit 42d5dd5

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

flagcx/adaptor/ccl/xccl_adaptor.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ BKCLOp flagcxRedOpToBKCLOp(flagcxRedOp_t op) {
4545
return BKCLOp::BKCL_MAX;
4646
case flagcxMin:
4747
return BKCLOp::BKCL_MIN;
48+
case flagcxAvg:
49+
return BKCLOp::BKCL_AVG;
4850
default:
4951
// return BKCLOp::BKCL_NUM_OPS to account for unknown redOp type
5052
return BKCLOp::BKCL_NUM_OPS;
@@ -363,4 +365,4 @@ struct flagcxCCLAdaptor xcclAdaptor = {
363365
xcclAdaptorDevCommReqsInit, xcclAdaptorDevCommCreate,
364366
xcclAdaptorDevCommDestroy};
365367

366-
#endif // USE_KUNLUNXIN_ADAPTOR
368+
#endif // USE_KUNLUNXIN_ADAPTOR

0 commit comments

Comments
 (0)