Skip to content

Add an upper bound for the video decoder workaround #3609

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.

Sign up for GitHub

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

Merged
JanuszL merged 2 commits into from
Jan 11, 2022
Merged

Add an upper bound for the video decoder workaround #3609

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
105af84
Add an upper bound for the video decoder workaround
JanuszL Jan 10, 2022
de215de
Review fix
JanuszL Jan 11, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 3 additions & 4 deletions dali/operators/reader/nvdecoder/nvdecoder.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,22 +59,21 @@ NvDecoder::NvDecoder(int device_id,
"Check your library paths and if NVIDIA driver is installed correctly.");


// This is a workaround for an issue with nvcuvid in drivers >460 where concurrent
// This is a workaround for an issue with nvcuvid in drivers >460 and < 470.21 where concurrent
// use on default context and non-default streams may lead to memory corruption.
// TODO(michalz): add an upper bound when the problem is fixed
bool use_default_stream = false;
#if NVML_ENABLED
{
nvml::Init();
static float driver_version = nvml::GetDriverVersion();
if (driver_version > 460)
if (driver_version > 460 && driver_version < 470.21)
use_default_stream = true;
}
#else
{
int driver_cuda_version = 0;
CUDA_CALL(cuDriverGetVersion(&driver_cuda_version));
if (driver_cuda_version >= 11030)
if (driver_cuda_version >= 11030 && driver_cuda_version < 11040)
use_default_stream = true;
}
#endif
Expand Down