[Bug] Incorrect handling of select dimension in torch frontend

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

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

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

評価

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

調査の方向性

Start in python/tvm/relax/frontend/torch/base_fx_graph_translator.py at the select handling near line 3129, then run the provided reproduction with PyTorch and the TVM backend. Compare the translated result for a[..., 0] with PyTorch's result; done when the selected dimension is removed and the example runs without ValueError.

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

説明

needs-triage type: bug

https://github.com/apache/tvm/blob/main/python/tvm/relax/frontend/torch/base_fx_graph_translator.py#L3129 tries to convert torch slice action to relax ops, but it does not remove the selected dimension.

Steps to reproduce
import torch
def foo(a, b):
    return a[..., 0] + b

print(foo(torch.randn(3, 2, 2), torch.randn(3, 2))) # valid
print(torch.compile(foo, backend = "tvm")(torch.randn(3, 2, 2), torch.randn(3, 2))) # will throw ValueError
主要言語
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 を短くまとめたダイジェスト。