saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-28 20:11:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix shape infer of onnx GroupNorm (#23477)

Browse source 
### 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")
This commit is contained in:
Yateng Hong 2025-01-26 15:41:29 +08:00 committed by GitHub
parent 1fc9c4823d
commit 97c2bbe3eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/symbolic_shape_infer.py
View file

@ -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",