Hacktoberfest 2026：メンテナが10月に向けて印を付けた、オープンで初心者向けの issue。 Hacktoberfest の issue を見る

[Vulkan] scalar_tensor uploads float bytes to integer scalar shaders

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

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

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

評価

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

調査の方向性

Start in backends/vulkan/runtime/graph/ops/impl/ScalarTensor.cpp and compare its parameter upload with the declaration in backends/vulkan/runtime/graph/ops/glsl/scalar_tensor.glsl. Run the provided Vulkan regression cases with the scalar-tensor exporter fixes from #23156. Done means the buffer type matches the shader declaration and integer outputs preserve values such as 16777217.

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

説明

module: vulkan
🐛 Describe the bug

While implementing the scalar-tensor delegation fixes in #23156, I found a separate runtime parameter-type mismatch in ScalarTensor.cpp.

The runtime always extracts the scalar as float and uploads a float parameter buffer, but selects the shader's scalar parameter type from graph.dtype_of(scalar_in). An integer literal therefore selects an integer parameter declaration while receiving float bytes. For example, 3.0f has the bit pattern 0x40400000, which an int32 parameter reads as 1077936128.

Converting every integer scalar through float also loses precision for exactly representable int32 values above 2**24, such as 16777217.

The relevant code is in ScalarTensor.cpp, and the parameter declaration comes from scalar_tensor.glsl.

This becomes reachable through ordinary exported models after fixing the raw ATen registration and serialized operator name described in item 2 of #23156. Without those exporter fixes, the scalar tensors fall back to CPU and hide this runtime defect.

The following regression cases exercise mixed scalar types and integer precision. They require a Vulkan-enabled runtime and the scalar-tensor exporter fixes from #23156:

import torch
from executorch.backends.vulkan.partitioner.vulkan_partitioner import VulkanPartitioner
from executorch.exir import to_edge_transform_and_lower
from executorch.extension.pybindings.portable_lib import (
    _load_for_executorch_from_buffer,
)


class WhereScalars(torch.nn.Module):
    def __init__(self, positive, negative):
        super().__init__()
        self.positive = positive
        self.negative = negative

    def forward(self, x):
        return torch.where(x, self.positive, self.negative)


inputs = (torch.tensor([True, False, True, False]),)
for positive, negative in ((3, -7.0), (3.0, -7), (16777217, -7)):
    model = WhereScalars(positive, negative)
    edge = to_edge_transform_and_lower(
        torch.export.export(model, inputs),
        partitioner=[VulkanPartitioner()],
    )
    program_buffer = edge.to_executorch().buffer
    module = _load_for_executorch_from_buffer(program_buffer)
    actual = module.run_method("forward", inputs)[0]
    torch.testing.assert_close(actual, model(*inputs), atol=0, rtol=0)

The parameter buffer's type must match the shader declaration. Integer outputs should preserve the integer value without an intermediate float conversion. A local fix using int32 parameters for integer outputs and float parameters for floating-point outputs passes all three cases on MoltenVK. The surrounding Vulkan regression suite also passes: 36 tests, including dynamic eager attention and SDPA.

Versions

ExecuTorch base commit 0c7ce72758c0cbb832c5ac5970626beb347ca745, with local changes for #23156; PyTorch 2.13.0; Python 3.11; macOS arm64 on Apple M1 Pro using MoltenVK.

Drafted with OpenAI Codex. The parameter-type diagnosis is based on the runtime source; the local fix and regression cases were executed on the GPU.

cc @SS-JIA @manuelcandales @digantdesai @cbilgin

主要言語
Python
スター
5k
フォーク
1.2k
平均マージ
2日 9時間
マージ済み PR（30日）
555

環境構築

はじめの一歩

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

pytorch/executorch のほかの issue

pytorch/executorch の issue をすべて見る

似ている issue

Python の issue をもっと見る

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

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