[X64] Fixed need_mask_register for eltwise injectors

openvinotoolkit · Aug 27, 2024 · c8ae8d9 · c8ae8d9
1 parent 234b379
commit c8ae8d9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/cpu/x64/injectors/jit_uni_eltwise_injector.cpp b/src/cpu/x64/injectors/jit_uni_eltwise_injector.cpp
@@ -1998,6 +1998,9 @@ bool jit_uni_eltwise_injector_f32<isa, Wmm>::need_mask_register(
             case eltwise_round: return false;
             case eltwise_hardswish: return false;
             case eltwise_hardsigmoid: return false;
+            case eltwise_hsigmoid: return false;
+            case eltwise_round_half_to_even: return false;
+            case eltwise_round_half_away_from_zero: return true;
             default: assert(!"unsupported eltwise algorithm");
         }
     } else {