-
Notifications
You must be signed in to change notification settings - Fork 609
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
Support for nvJPEG preallocate API for batched HW decoder #2544
Conversation
!build |
CI MESSAGE: [1895377]: BUILD STARTED |
CI MESSAGE: [1895377]: BUILD FAILED |
CI MESSAGE: [1895383]: BUILD STARTED |
CI MESSAGE: [1895383]: BUILD FAILED |
21415d8
to
f7dc868
Compare
CI MESSAGE: [1899975]: BUILD STARTED |
CI MESSAGE: [1899977]: BUILD STARTED |
CI MESSAGE: [1899975]: BUILD PASSED |
CI MESSAGE: [1899977]: BUILD PASSED |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
NVJPEG_CALL(nvjpegDecodeBatchedPreAllocate( | ||
handle_, | ||
state_hw_batched_, | ||
hw_decoder_bs_, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this might collide with the ongoing variable batch size changes. Maybe check with @szalpal
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Synced offline. Should be fine.
.AddOptionalArg("preallocate_width_hint", | ||
R"code(Applies **only** to the ``mixed`` backend type. | ||
|
||
Hint about the expected maximum width of the processed images. It helps to allocate ahead of time |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Image width hint.
The hint is used to preallocate memory for the HW JPEG decoder in NVIDIA Ampere GPU architecture.
maybe? (same for height)
Signed-off-by: Albert Wolant <awolant@nvidia.com>
500c02d
to
3d89f34
Compare
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
e88778c
to
a6f768d
Compare
!build |
CI MESSAGE: [1900628]: BUILD STARTED |
CI MESSAGE: [1900628]: BUILD PASSED |
Signed-off-by: Albert Wolant awolant@nvidia.com
Why we need this PR?
Pick one, remove the rest
What happened in this PR?
Fill relevant points, put NA otherwise. Replace anything inside []
adds support for nvJPEG preallocate API for batched HW decoder
ImageDecoder
NA
CI
Operator docs are updated
JIRA TASK: [DALI-1443]