-
Notifications
You must be signed in to change notification settings - Fork 607
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
Enable CUDA 11.8 and Hopper support #4308
Conversation
CI MESSAGE: [6085306]: BUILD STARTED |
@@ -115,6 +118,14 @@ class nvJPEGDecoder : public Operator<MixedBackend>, CachedDecoderImpl { | |||
#endif | |||
LOG_LINE << "Using NVJPEG_BACKEND_HARDWARE" << std::endl; | |||
CUDA_CALL(nvjpegJpegStateCreate(handle_, &state_hw_batched_)); | |||
if (nvjpegIsSymbolAvailable("nvjpegGetHardwareDecoderInfo")) { | |||
nvjpegGetHardwareDecoderInfo(handle_, &num_hw_engines_, &num_hw_cores_per_engine_); | |||
// ToDo adjust hw_decoder_load_ based on num_hw_engines_ and num_hw_cores_per_engine_ |
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.
// ToDo adjust hw_decoder_load_ based on num_hw_engines_ and num_hw_cores_per_engine_ | |
// TODO(jlisiecki) adjust hw_decoder_load_ based on num_hw_engines_ and num_hw_cores_per_engine_ |
won't the linter complain here?
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.
Apparently not
CI MESSAGE: [6086992]: BUILD STARTED |
CI MESSAGE: [6085306]: BUILD PASSED |
CI MESSAGE: [6086992]: BUILD FAILED |
- adds CUDA 11.8 based build - extend the image decoder to support H100 Hopper Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
CI MESSAGE: [6096383]: BUILD STARTED |
CI MESSAGE: [6096383]: BUILD PASSED |
- adds CUDA 11.8 based build - extend the image decoder to support H100 Hopper Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
- adds CUDA 11.8 based build - extend the image decoder to support H100 Hopper Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
Signed-off-by: Janusz Lisiecki jlisiecki@nvidia.com
Category:
Other (e.g. Documentation, Tests, Configuration)
Description:
Additional information:
Affected modules and functionalities:
Key points relevant for the review:
Tests:
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: N/A