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

Remove explicit batch network flag for TRT 10+ #24298

Merged
merged 1 commit into from
Apr 8, 2025

Conversation

kevinch-nv
Copy link
Contributor

Description

nvinfer1::NetworkDefinitionCreationFlag::kEXPLICIT_BATCH has been deprecated since 10.0 and is always implicitly set for versions 10.0+. Change the EP code to only set this flag for TRT versions 8 and below.

Motivation and Context

Removes deprecated API usages in the TRT EP code.

Signed-off-by: Kevin Chen <kevinch@nvidia.com>
@jywu-msft jywu-msft requested a review from yf711 April 7, 2025 16:56
@yf711
Copy link
Contributor

yf711 commented Apr 7, 2025

/azp run Linux QNN CI Pipeline, Win_TRT_Minimal_CUDA_Test_CI, Windows ARM64 QNN CI Pipeline, Windows GPU Doc Gen CI Pipeline, Windows x64 QNN CI Pipeline

Copy link

Azure Pipelines successfully started running 5 pipeline(s).

@jywu-msft jywu-msft requested a review from chilo-ms April 7, 2025 16:59
Copy link
Contributor

@yf711 yf711 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@yf711 yf711 merged commit 4edada6 into microsoft:main Apr 8, 2025
69 checks passed
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

Successfully merging this pull request may close these issues.

3 participants