Make nvjpeg handle creation once_flag static (#5713)

6db54fb7 · Nicolas Hug · GitHub · 88aaf51e · 6db54fb7
Unverified Commit 6db54fb7 authored Mar 31, 2022 by Nicolas Hug Committed by GitHub Mar 31, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/csrc/io/image/cuda/decode_jpeg_cuda.cpp torchvision/csrc/io/image/cuda/decode_jpeg_cuda.cpp +1 -1

No files found.
--- a/torchvision/csrc/io/image/cuda/decode_jpeg_cuda.cpp
+++ b/torchvision/csrc/io/image/cuda/decode_jpeg_cuda.cpp
@@ -71,7 +71,7 @@ torch::Tensor decode_jpeg_cuda(
  at::cuda::CUDAGuard device_guard(device);

  // Create global nvJPEG handle
-  std::once_flag nvjpeg_handle_creation_flag;
+  static std::once_flag nvjpeg_handle_creation_flag;
  std::call_once(nvjpeg_handle_creation_flag, []() {
    if (nvjpeg_handle == nullptr) {
      nvjpegStatus_t create_status = nvjpegCreateSimple(&nvjpeg_handle);