Skip to content

add support for TARGET_DEVICE parameter #87

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

Open
wants to merge 18 commits into
base: main
Choose a base branch
from

Conversation

dtrawins
Copy link
Collaborator

This is updated version of #74 which doesn't require any changed on the base triton image.
It gives option to specify TARGET_DEVICE to be passed to the OpenVINO Runtime during model compilation

@dtrawins dtrawins requested review from tanmayv25 and nnshah1 December 4, 2024 12:58
Comment on lines +321 to +323
std::transform(
value.begin(), value.end(), value.begin(),
[](unsigned char c) { return std::toupper(c); });
Copy link
Collaborator

Choose a reason for hiding this comment

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

Suggested change
std::transform(
value.begin(), value.end(), value.begin(),
[](unsigned char c) { return std::toupper(c); });
std::transform(value.begin(), value.end(), value.begin(), std::toupper);

{
if (Kind() != TRITONSERVER_INSTANCEGROUPKIND_CPU) {
throw triton::backend::BackendModelInstanceException(TRITONSERVER_ErrorNew(
TRITONSERVER_ERROR_INVALID_ARG,
(std::string("unable to load model '") + model_state_->Name() +
"', Triton openVINO backend supports only CPU device")
"', Triton OpenVINO backend supports only Kind CPU and AUTO")
Copy link
Collaborator

Choose a reason for hiding this comment

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

Wouldn't we exit here for target device = GPU?

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

Successfully merging this pull request may close these issues.

2 participants