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

Support for nvJPEG preallocate API for batched HW decoder #2544

Merged
merged 3 commits into from
Dec 14, 2020

Conversation

JanuszL
Copy link
Contributor

@JanuszL JanuszL commented Dec 11, 2020

Signed-off-by: Albert Wolant awolant@nvidia.com

Why we need this PR?

Pick one, remove the rest

  • It adds support for nvJPEG preallocate API for batched HW decoder

What happened in this PR?

Fill relevant points, put NA otherwise. Replace anything inside []

  • What solution was applied:
    adds support for nvJPEG preallocate API for batched HW decoder
  • Affected modules and functionalities:
    ImageDecoder
  • Key points relevant for the review:
    NA
  • Validation and testing:
    CI
  • Documentation (including examples):
    Operator docs are updated

JIRA TASK: [DALI-1443]

@JanuszL
Copy link
Contributor Author

JanuszL commented Dec 11, 2020

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1895377]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1895377]: BUILD FAILED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1895383]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1895383]: BUILD FAILED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1899975]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1899977]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1899975]: BUILD PASSED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1899977]: BUILD PASSED

Copy link
Collaborator

@banasraf banasraf left a 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_,
Copy link
Contributor

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

Copy link
Contributor Author

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
Copy link
Contributor

@jantonguirao jantonguirao Dec 14, 2020

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)

awolant and others added 2 commits December 14, 2020 16:46
Signed-off-by: Albert Wolant <awolant@nvidia.com>
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
@JanuszL
Copy link
Contributor Author

JanuszL commented Dec 14, 2020

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1900628]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1900628]: BUILD PASSED

@JanuszL JanuszL merged commit afbccb9 into NVIDIA:master Dec 14, 2020
@JanuszL JanuszL deleted the use_memory_pool_hw branch December 14, 2020 21:39
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.

None yet

5 participants