[Bug] [Relax][ONNX] Elu, Selu and LRN construct FLOAT scalars for FLOAT16 inputs

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

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

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

評価

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

調査の方向性

3つのコンバーターは、Relax ONNX フロントエンドにある Elu、Selu、LRN のハンドラーで、FLOAT スカラー定数が構築される箇所です。Elu と Selu は減算用のデフォルト FLOAT 定数を作成し、LRN は乗算用の FLOAT 定数を作成します。まず issue の再現スクリプトを実行し、FLOAT と FLOAT16 のパスを比較してください。3つの op がすべて FLOAT16 でインポートでき、スカラー定数が入力の dtype と一致し、FLOAT のケースも引き続き成功すれば完了です。

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

説明

needs-triage type: bug
Expected behavior

The ONNX Elu, Selu and LRN converters should accept FLOAT16 inputs, as allowed by their schemas, and create scalar constants with a compatible dtype.

Actual behavior

All three FLOAT controls import. The otherwise identical FLOAT16 models pass full ONNX checking but fail with:

TypeError: Binary operators must have the same datatype for both operands.

Elu/Selu construct default FLOAT constants for subtraction; LRN explicitly creates FLOAT constants for multiplication. This is in the Relax ONNX converters, separate from the earlier TOPI LRN result-dtype fix in #11032.

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 onnx
from onnx import TensorProto as T, helper as h
from tvm.relax.frontend.onnx import from_onnx

for op in ["Elu", "Selu", "LRN"]:
    for dtype in [T.FLOAT, T.FLOAT16]:
        attrs = {"size": 3} if op == "LRN" else {}
        node = h.make_node(op, ["x"], ["y"], **attrs)
        model = h.make_model(h.make_graph([node], op,
            [h.make_tensor_value_info("x", dtype, [1, 3, 2, 2])],
            [h.make_tensor_value_info("y", dtype, [1, 3, 2, 2])]),
            opset_imports=[h.make_opsetid("", 18)], ir_version=10)
        onnx.checker.check_model(model, full_check=True)
        try:
            from_onnx(model, keep_params_in_input=False)
            print(op, T.DataType.Name(dtype), "imported")
        except Exception as error:
            print(op, T.DataType.Name(dtype), type(error).__name__, str(error).splitlines()[-1])
Triage
  • needs-triage
主要言語
Python
スター
13.9k
フォーク
4k
平均マージ
8時間 2分
マージ済み PR（30日）
180

環境構築

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

はじめの一歩

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

apache/tvm のほかの issue

apache/tvm の issue をすべて見る

似ている issue

Python の issue をもっと見る

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

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