[Bug] [Relax][ONNX] Constant INT64 Concat data becomes ShapeExpr and breaks Add

オープン
  • 何か問題がありますか？
#20,889 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る

メンテナーはふだん 1 日以内に返信

まだ誰も着手していません。

評価

難易度
4/5
見積もり時間
1〜2日
初心者へのやさしさ
38/100
issue の種類
バグ
明瞭さ
明確に書かれている
活発さ
活発
技術スタック
python
領域
compilers

調査の方向性

The failure starts in the Relax ONNX frontend entry point from_onnx when a Concat of two constant INT64 tensors feeds an Add node named add_data, which raises "cannot handle ShapeExpr inputs". Run the reproduction script from the issue to confirm that only the all-constant variant fails, then trace where the Concat output is classified as a shape rather than tensor data. Done means both variants import and the Add returns [11, 22], matching the ReferenceEvaluator result.

索引モデルが issue の本文から書いたものです。

説明

needs-triage type: bug
Expected behavior

Concatenating INT64 tensors [1] and [2], then adding [10, 20], should return the INT64 tensor [11, 22], whether the operands are constants or runtime inputs.

Actual behavior

Making both Concat operands constant causes the frontend to treat ordinary tensor data as a shape. Its Add consumer fails:

ValueError: Node add_data cannot handle ShapeExpr inputs.

Keeping one operand as runtime data imports successfully. Both ONNX variants pass full checking and reference evaluation. Unlike the mixed shape/tensor case in #19498, this graph does not construct a shape.

Environment

TVM 0.28.dev0 (local build; build commit unavailable), Python 3.12.3, Linux x86_64; ONNX 1.22.0, NumPy 2.5.4.

Steps to reproduce

Run with python repro.py:

import numpy as np
import onnx
from onnx import TensorProto as T, helper as h, numpy_helper as nh
from onnx.reference import ReferenceEvaluator
from tvm.relax.frontend.onnx import from_onnx

for constant_a in [False, True]:
    nodes = [h.make_node("Constant", [], ["b"], value=nh.from_array(np.array([2], dtype="int64")))]
    inputs = [h.make_tensor_value_info("x", T.INT64, [2])]
    feeds = {"x": np.array([10, 20], dtype="int64")}
    if constant_a:
        nodes.append(h.make_node("Constant", [], ["a"], value=nh.from_array(np.array([1], dtype="int64"))))
    else:
        inputs.append(h.make_tensor_value_info("a", T.INT64, [1]))
        feeds["a"] = np.array([1], dtype="int64")
    nodes += [h.make_node("Concat", ["a", "b"], ["c"], axis=0),
              h.make_node("Add", ["c", "x"], ["y"], name="add_data")]
    model = h.make_model(h.make_graph(nodes, "concat", inputs,
        [h.make_tensor_value_info("y", T.INT64, [2])]),
        opset_imports=[h.make_opsetid("", 18)], ir_version=10)
    onnx.checker.check_model(model, full_check=True)
    np.testing.assert_array_equal(ReferenceEvaluator(model).run(None, feeds)[0], [11, 22])
    try:
        from_onnx(model, keep_params_in_input=False)
        print("constant_a", constant_a, "imported")
    except Exception as error:
        print("constant_a", constant_a, type(error).__name__, str(error).splitlines()[-1])
Triage
  • needs-triage
主要言語
Python
スター
13.9k
フォーク
4k
平均マージ
7時間 59分
マージ済み PR（30日）
159

環境構築

このプロジェクトには開発コンテナ、Dockerfile、コントリビューションガイドがありません。まず README を読み、一般的な手順ははじめてのコントリビューションガイドを参照してください。

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

apache/tvm のほかの issue

apache/tvm の issue をすべて見る

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。