1 parent fed5532 commit 42d5dd5Copy full SHA for 42d5dd5
1 file changed
flagcx/adaptor/ccl/xccl_adaptor.cc
@@ -45,6 +45,8 @@ BKCLOp flagcxRedOpToBKCLOp(flagcxRedOp_t op) {
45
return BKCLOp::BKCL_MAX;
46
case flagcxMin:
47
return BKCLOp::BKCL_MIN;
48
+ case flagcxAvg:
49
+ return BKCLOp::BKCL_AVG;
50
default:
51
// return BKCLOp::BKCL_NUM_OPS to account for unknown redOp type
52
return BKCLOp::BKCL_NUM_OPS;
@@ -363,4 +365,4 @@ struct flagcxCCLAdaptor xcclAdaptor = {
363
365
xcclAdaptorDevCommReqsInit, xcclAdaptorDevCommCreate,
364
366
xcclAdaptorDevCommDestroy};
367
-#endif // USE_KUNLUNXIN_ADAPTOR
368
+#endif // USE_KUNLUNXIN_ADAPTOR
0 commit comments