From 97c2bbe3eb676d443d92e14447d4c60d92757530 Mon Sep 17 00:00:00 2001
From: Yateng Hong <yatengh@microsoft.com>
Date: Sun, 26 Jan 2025 15:41:29 +0800
Subject: [PATCH] Fix shape infer of onnx GroupNorm (#23477)

### Description
<!-- Describe your changes. -->
Fix shape infer of onnx GroupNorm.


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

Unable to run shape inference for onnx `GroupNorm`.


[model.onnx](https://raw.githubusercontent.com/onnx/onnx/refs/heads/main/onnx/backend/test/data/node/test_group_normalization_example/model.onnx)

> python
D:\source\cognition\onnxruntime\onnxruntime\python\tools\symbolic_shape_infer.py
--input model.onnx
Traceback (most recent call last):
File
"D:\source\cognition\onnxruntime\onnxruntime\python\tools\symbolic_shape_infer.py",
line 2999, in <module>
    out_mp = SymbolicShapeInference.infer_shapes(
File
"D:\source\cognition\onnxruntime\onnxruntime\python\tools\symbolic_shape_infer.py",
line 2935, in infer_shapes
    raise Exception("Incomplete symbolic shape inference")
---
 onnxruntime/python/tools/symbolic_shape_infer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/onnxruntime/python/tools/symbolic_shape_infer.py b/onnxruntime/python/tools/symbolic_shape_infer.py
index 7a6028dfbe..b9675d4280 100755
--- a/onnxruntime/python/tools/symbolic_shape_infer.py
+++ b/onnxruntime/python/tools/symbolic_shape_infer.py
@@ -205,6 +205,7 @@ class SymbolicShapeInference:
             "GemmFastGelu": self._infer_GemmFastGelu,
             "GemmFloat8": self._infer_GemmFloat8,
             "GroupNorm": self._infer_GroupNorm,
+            "GroupNormalization": self._infer_GroupNorm,
             "GroupQueryAttention": self._infer_GroupQueryAttention,
             "LayerNormalization": self._infer_LayerNormalization,
             "LongformerAttention": self._infer_LongformerAttention,
@@ -474,6 +475,7 @@ class SymbolicShapeInference:
             "PythonOp",
             "MultiHeadAttention",
             "GroupNorm",
+            "GroupNormalization",
             "GroupQueryAttention",
             "SparseAttention",
             "SkipGroupNorm",