Skip to content

onnx.checker crashes on STFT #6186

@mgehre-amd

Description

@mgehre-amd

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions