cpu: x64: rnn: enable merged layer computation for lbr_gru

asimonov1 · asimonov1 · commit 5a86c1f0c4ed · 2025-07-17T22:44:58.000+02:00
diff --git a/src/cpu/x64/rnn/rnn_brgemm_utils.cpp b/src/cpu/x64/rnn/rnn_brgemm_utils.cpp
@@ -499,8 +499,10 @@ status_t rnn_brgemm_t<prop_kind::forward>::configure_brgemm(
 
     // enable merged across n_iter dimension layer part of the cell computation
     // TODO: extend coverage for other problem types
-    const bool mlc_cell_type_ok = cell_kind == alg_kind::vanilla_lstm
-            && !rnn.is_lstm_projection && !rnn.is_lstm_peephole;
+    const bool mlc_cell_type_ok
+            = (cell_kind == alg_kind::vanilla_lstm && !rnn.is_lstm_projection
+                      && !rnn.is_lstm_peephole)
+            || (cell_kind == alg_kind::lbr_gru && rnn.brgemm_isa == x64::avx2);
     const int mlc_mb_max_threshold = 1;
     const int mlc_n_iter_min_threshold = 2;
     const int mlc_n_layer_max_threshold = 1;
