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

Fix shape and reformat free tensor handling in the input byte size check #380

Merged
merged 19 commits into from
Jul 27, 2024

Conversation

pskiran1
Copy link
Member

@pskiran1 pskiran1 commented Jul 14, 2024

  • Added support for the is_non_linear_format_io flag in the model configuration to identify if input/output tensors use a non-linear IO format.
  • For shape tensors, use the original shape for byte size validation.
  • For tensors using the non-linear IO format (reformat-free tensors), validate the input byte size in the backend.

src/model_config_utils.cc Outdated Show resolved Hide resolved
src/model_config_utils.cc Outdated Show resolved Hide resolved
src/infer_request.cc Outdated Show resolved Hide resolved
src/infer_request.cc Show resolved Hide resolved
src/infer_request.h Outdated Show resolved Hide resolved
@pskiran1 pskiran1 requested review from tanmayv25 and GuanLuo July 18, 2024 12:14
@pskiran1 pskiran1 requested a review from tanmayv25 July 18, 2024 18:59
tanmayv25
tanmayv25 previously approved these changes Jul 18, 2024
@pskiran1 pskiran1 requested a review from tanmayv25 July 22, 2024 14:33
@pskiran1 pskiran1 merged commit 9ed1544 into main Jul 27, 2024
1 check passed
@pskiran1 pskiran1 deleted the spolisetty_dlis_6805 branch July 30, 2024 11:47
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.

3 participants