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

[TensorRT EP] Avoid calling unavailable function with cpu python package #19251

Merged
merged 2 commits into from
Jan 24, 2024

Conversation

chilo-ms
Copy link
Contributor

C.register_tensorrt_plugins_as_custom_ops() is only available in gpu python package.
Add condition to avoid calling it in cpu python package.

jywu-msft
jywu-msft previously approved these changes Jan 24, 2024
@jywu-msft jywu-msft merged commit c10be18 into main Jan 24, 2024
96 of 98 checks passed
@jywu-msft jywu-msft deleted the chi/python_trt_ep_fix branch January 24, 2024 05:30
YUNQIUGUO pushed a commit that referenced this pull request Jan 25, 2024
…age (#19251)

C.register_tensorrt_plugins_as_custom_ops() is only available in gpu
python package.
Add condition to avoid calling it in cpu python package.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants