Skip to content

PP-FormulaNet_plus-S onnx error on gpu #1593

@ningpp

Description

@ningpp

CPU多次推理都是正确的。
GPU第一次推理是正确的,返回的结果ndarray的shape是(1, 51)
GPU后面几次推理结果的是错误的,返回的结果ndarray的shape是(1, 1026)

依赖

onnx                     1.17.0
onnx_graphsurgeon        0.5.8
onnxoptimizer            0.3.13
onnxruntime-gpu          1.22.0
opt-einsum               3.3.0
paddle2onnx              2.0.2rc3
paddlepaddle-gpu         3.0.0.dev20250617
paddlex                  3.0.2

测试用的图片

Image

复现代码

ningpp/tmp-paddlex-onnx

PaddlePaddle/PaddleX#4238

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions