fix

Author: xadupre

_unittests/ut_helpers/test_log_helper.py

@@ -268,7 +268,7 @@ def test_cube_logs_performance_cube_time(self):
         cube = CubeLogsPerformance(dfs, keep_last_date=True)
         cube.load()
         ct = cube.clone()
-        self.assertEqual((52, 111), ct.shape)
+        self.assertEqual((52, 116), ct.shape)
 
     def test_duplicate(self):
         df = pandas.DataFrame(

onnx_diagnostic/helpers/log_helper.py

@@ -1517,6 +1517,7 @@ def __init__(
             "n_node_control_flow",
             "n_node_expand",
             "n_node_function",
+            "n_node_gqa",
             "n_node_initializer",
             "n_node_initializer_small",
             "n_node_layer_normalization",
@@ -1727,6 +1728,11 @@ def first_err(df: pandas.DataFrame) -> pandas.Series:
                     + gdf(df, "op_onnx_com.microsoft_DecoderMaskedMultiHeadAttention", 0)
                     + gdf(df, "op_onnx_com.microsoft_SparseAttention", 0),
                 ),
+                n_node_gqa=lambda df: gpreserve(
+                    df,
+                    "time_latency_eager",
+                    gdf(df, "op_onnx_com.microsoft_GroupQueryAttention", 0),
+                ),
                 n_node_layer_normalization=lambda df: gpreserve(
                     df,
                     "time_latency_eager",