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

getBindingIndex() is DEPRECATED #3746

Closed
Egorundel opened this issue Mar 28, 2024 · 3 comments
Closed

getBindingIndex() is DEPRECATED #3746

Egorundel opened this issue Mar 28, 2024 · 3 comments
Labels
triaged Issue has been triaged by maintainers

Comments

@Egorundel
Copy link

Description

Hello!
I want to find out the BindingIndex by the tensor name.
How can I do this, provided that the getBindingIndex() method is deprecated in 8.5.

Environment

TensorRT Version: 8.6

NVIDIA GPU: RTX 3060

NVIDIA Driver Version: 545.23.06

CUDA Version: 11.1

CUDNN Version: 8.0.4

Operating System:

Python Version (if applicable): 3.8

PyTorch Version (if applicable): 10.1

@lix19937
Copy link

lix19937 commented Mar 28, 2024

If you just want to get shape(dims) of tensor(binding), use getTensorShape(char const * tensorName), when TensorRT Version is 8.6

Ref https://docs.nvidia.com/deeplearning/tensorrt/api/c_api/classnvinfer1_1_1_i_cuda_engine.html#af96a2ee402ab47b7e0b7f0becb63d693

@zerollzeng zerollzeng added the triaged Issue has been triaged by maintainers label Mar 31, 2024
@DeepLearnerYe
Copy link

getTensorShape(char const * tensorName) requires tensorName. If I want to get the name automatically instead of specifying the name, getBindingName(int32_t) is provided. But it is also deprecated in 8.5, what can I do?

@lix19937
Copy link

@DeepLearnerYe
Use

int32_t nvinfer1::ICudaEngine::getNbIOTensors()	
char const * nvinfer1::ICudaEngine::getIOTensorName(int32_t index) const

ref in https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-853/api/index.html

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
triaged Issue has been triaged by maintainers
Projects
None yet
Development

No branches or pull requests

4 participants