[Accuracy] PersonaPlex FP16 residuals skip linear-output rounding
メンテナーはふだん 1 日以内に返信
@chaofengw-nv がすでに取り組んでいます。
2026年10月18日 から。
評価
この issue はまだ評価されていません。
説明
Description
PersonaPlex's FP16 decoder residual path can add the unrounded GEMM accumulator instead of first rounding the linear output to Half, as the Native model does. This changes values inside an actual decoder layer, before subsequent normalization and autoregressive generation.
Affected base: 9a8b545f5e8c22c653c8029c95a999dc070b4f3f.
Reproduction
Use a Half projection whose exact dot product is 1 + 2^-12, followed by a residual of -1:
- Native: round the linear output to Half (
1), then add the Half residual:0. - Current TensorRT decoder:
2^-12(0.000244140625), consistent with fusion across the linear-output rounding boundary.
A regression invokes the existing default_decoder._add_decoder_layer with the depth dimensions (1024 hidden channels, 16 heads, cache length 18, MLP width 2816). It uses dense binary weights, one valid attention key and a zero MLP, and reads the layer's normal hidden and present-K/V outputs. The Native CUDA linear operations independently establish the expected scalar.
On GB300, TensorRT 11.2.0.113, CUDA 13 and PyTorch 2.12.0+cu130, three independent original Half engines each return 2^-12 instead of zero. All three original FP32 control engines pass; neither precision has errors or skips.
A separate fresh-process probe also reproduces the discrepancy at widths 1024 and 4096, three builds each. Performing the residual sum in FP32 after converting the already-Half branch outputs, then restoring Half, returns the Native result in all six cases.
Expected behavior
Preserve the model-dtype linear-output rounding boundary before the residual sum. Keep Half weights, hidden states and KV caches, and retain the existing FP32 behavior. The Native sequential attention and MLP residuals are defined in the pinned official transformer implementation.
This is a component reproduction. Full speech-benchmark acceptance remains separate; no benchmark input, Native baseline or acceptance gate is changed.
- 主要言語
- Python
- スター
- 276
- フォーク
- 73
- 平均マージ
- 1日 9時間
- マージ済み PR（30日）
- 156
環境構築
- Dockerfile または Docker Compose ファイルあり
- プルリクエストのテンプレートあり
- コントリビューションガイドを読む
はじめの一歩
- issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
- 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
- リポジトリをフォークし、ブランチを切って変更します。
- issue 番号を参照したプルリクエストを送ります。
NVIDIA/TensorRT-Model-Connect のほかの issue
-
Docs: ITrtModule::forward() output-buffer lifetime and dtype contract are undocumented再び着手できるかも このイシューのプルリクエストはマージされずにクローズされました。 オープン
難易度 2/5 1〜3時間 初心者へのやさしさ 82/100
NVIDIA/TensorRT-Model-Connect#1126 · コメント 1 件 ·
メンテナーはふだん 1 日以内に返信
-
[Accuracy] MoGe selected convolutions replace Native replicate padding with zeros対応中かも @chaofengw-nv が 7 日前に担当しました。 オープンbug
NVIDIA/TensorRT-Model-Connect#1727 · 担当者 1 名 ·
メンテナーはふだん 1 日以内に返信
-
難易度 5/5 1週間以上 初心者へのやさしさ 6/100
NVIDIA/TensorRT-Model-Connect#1700 · コメント 4 件 ·
メンテナーはふだん 1 日以内に返信
-
Feature: Nomic Embed Text v1.5 native embeddings対応中かも @JiaxinD が 19 日前に担当しました。 オープン
難易度 5/5 1週間以上 初心者へのやさしさ 12/100
NVIDIA/TensorRT-Model-Connect#1582 ·
メンテナーはふだん 1 日以内に返信
-
retry_backoff retries the entire multi-family GPU test run even when a single family's genuine (non-transient) failure is the cause対応中かも このイシューにリンクされたプルリクエストがオープン中、またはマージ済みです。 オープン
難易度 4/5 3〜5日 初心者へのやさしさ 48/100
NVIDIA/TensorRT-Model-Connect#1576 ·
メンテナーはふだん 1 日以内に返信
NVIDIA/TensorRT-Model-Connect の issue をすべて見る
似ている issue
-
Python Volcengine provider returns failures instead of raising them対応中かも @Ricardo-M-L が今日担当しました。 オープン
難易度 2/5 1〜3時間 初心者へのやさしさ 79/100
agent-infra/sandbox#248 ·
-
Textractor leaves empty temporary files when remote retrieval fails対応中かも @HuaTNA が今日担当しました。 オープン
難易度 2/5 1〜3時間 初心者へのやさしさ 82/100
メンテナーはふだん 1 日以内に返信
-
test(bm_generation): pin the learning parameters and CPU sync shortcut対応中かも @iloveeyjafjalla が今日担当しました。 オープン
難易度 2/5 1〜3時間 初心者へのやさしさ 78/100
qboson/kaiwu-pytorch-plugin#1341 ·
メンテナーはふだん 2 日以内に返信
-
難易度 2/5 1〜3時間 初心者へのやさしさ 70/100
elder-plinius/V3R1T4S#625 ·
メンテナーはふだん 1 日以内に返信
-
難易度 2/5 1〜3時間 初心者へのやさしさ 62/100
foxglove/mcap#1907 · コメント 1 件 ·
メンテナーはふだん 1 日以内に返信