Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Handle duplicate constant names in local functions #974

Open
olesalscheider opened this issue Jun 4, 2024 · 0 comments
Open

Handle duplicate constant names in local functions #974

olesalscheider opened this issue Jun 4, 2024 · 0 comments

Comments

@olesalscheider
Copy link

Description

The dynamo based ONNX export of pytorch 2 can create ONNX graphs with local functions where the names of constants are only unique within the function but not within the whole ONNX model (it seems).

This leads to weights being (wrongly) reused. Amongst other things, it can cause this error message

[TRT] [E] [network.cpp::setWeightsName::3797] Error Code 1: Internal Error (Error: Weights of same values but of different counts are used in the network!)

I think the correct solution might be to prefix all weights with the name of the local function?

Environment

TensorRT Version: 10.0.1.6
ONNX-TensorRT Version / Branch: main
GPU Type: RTX A6000
Nvidia Driver Version: 550
CUDA Version: 12.4
CUDNN Version: 9
Operating System + Version: Ubuntu 22.04
Python Version (if applicable): 3.10.12
TensorFlow + TF2ONNX Version (if applicable): -
PyTorch Version (if applicable): nightly
Baremetal or Container (if container which image + tag): baremetal

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant