fix(gopt): fix replace-reduce-opr when dump io16c32

GitOrigin-RevId: b52a08a9c07977bce38f6a2ac3aa5aa68795f16b
MegEngine · Dec 11, 2023 · 5c0e1e7 · 5c0e1e7
1 parent 822a8cb
commit 5c0e1e7
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/gopt/impl/inference.cpp b/src/gopt/impl/inference.cpp
@@ -898,20 +898,21 @@ std::unique_ptr<ConvertF32ToF16Pass> ConvertF32ToF16Pass::make(bool use_f32_comp
                                       const VarNodeArray& new_inp) {
         auto& reduce_opr = opr->cast_final_safe<opr::Reduce>();
         auto new_param = reduce_opr.param();
-        if (use_f32_comp) {
+        if (use_f32_comp &&
+            opr->input(0)->dtype().category() == megdnn::DTypeCategory::FLOAT) {
             new_param.data_type = megdnn::param::Reduce::DataType::FLOAT_O16xC32;
         }
         if (opr->input().size() == 1) {
-            auto new_matmul_opr =
+            auto new_reduce_opr =
                     opr::Reduce::make(new_inp[0], new_param, {}, reduce_opr.config());
-            return new_matmul_opr.node()->owner_opr();
+            return new_reduce_opr.node()->owner_opr();
         } else {
             mgb_assert(
                     opr->input().size() == 2, "invalid input size %zu",
                     opr->input().size());
-            auto new_matmul_opr = opr::Reduce::make(
+            auto new_reduce_opr = opr::Reduce::make(
                     new_inp[0], new_param, new_inp[1], reduce_opr.config());
-            return new_matmul_opr.node()->owner_opr();
+            return new_reduce_opr.node()->owner_opr();
         }
     };