-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Closed
Labels
Description
Bug Report
Describe the bug
#!/usr/bin/env python
import onnx
import onnx.parser
modeltxt = """
<
ir_version: 10,
opset_import: ["" : 17]
>
graph (float16[3] in0, int32[2] in1, float16[7,8,8,8] in2, int32[8,1,7,2] in3) => () {
out0 = STFT (in0, in1, in2, in3)
}
"""
model = onnx.parser.parse_model(modeltxt)
print("Parser: Passed")
onnx.checker.check_model(model, full_check=True)
print("Checker: Passed")
onnx.shape_inference.infer_shapes(model)
crashes with
Parser: Passed
[1] 187375 segmentation fault ./repro_bug.py
System information
- OS Platform and Distribution (. Linux Ubuntu 20.04): Linux Ubuntu 20.04
- ONNX version (e.g. 1.13): 1.16.1
- Python version: 3.10.12
Expected behavior
No crash, but an error that the model is invalid