Linter

Author: NicolasHug

src/torchcodec/_core/CudaDeviceInterface.cpp

@@ -226,36 +226,37 @@ void CudaDeviceInterface::convertAVFrameToFrameOutput(
 
   cudaStream_t rawStream = at::cuda::getCurrentCUDAStream().stream();
 
-  // Build an NppStreamContext, either via the old helper or by hand on CUDA 12.9+
+  // Build an NppStreamContext, either via the old helper or by hand on
+  // CUDA 12.9+
   NppStreamContext nppCtx{};
-  #if CUDA_VERSION < 12090
-    NppStatus ctxStat = nppGetStreamContext(&nppCtx);
-    TORCH_CHECK(ctxStat == NPP_SUCCESS, "nppGetStreamContext failed");
-    // override if you want to force a particular stream
-    nppCtx.hStream = rawStream;
-  #else
-    // CUDA 12.9+: helper was removed, we need to build it manually
-    int dev = 0;
-    cudaError_t err = cudaGetDevice(&dev);
-    TORCH_CHECK(err == cudaSuccess, "cudaGetDevice failed");
-    cudaDeviceProp prop{};
-    err = cudaGetDeviceProperties(&prop, dev);
-    TORCH_CHECK(err == cudaSuccess, "cudaGetDeviceProperties failed");
-
-    nppCtx.nCudaDeviceId                       = dev;
-    nppCtx.nMultiProcessorCount                = prop.multiProcessorCount;
-    nppCtx.nMaxThreadsPerMultiProcessor        = prop.maxThreadsPerMultiProcessor;
-    nppCtx.nMaxThreadsPerBlock                 = prop.maxThreadsPerBlock;
-    nppCtx.nSharedMemPerBlock                  = prop.sharedMemPerBlock;
-    nppCtx.nCudaDevAttrComputeCapabilityMajor  = prop.major;
-    nppCtx.nCudaDevAttrComputeCapabilityMinor  = prop.minor;
-    nppCtx.nStreamFlags                        = 0;
-    nppCtx.hStream                             = rawStream;
-  #endif
+#if CUDA_VERSION < 12090
+  NppStatus ctxStat = nppGetStreamContext(&nppCtx);
+  TORCH_CHECK(ctxStat == NPP_SUCCESS, "nppGetStreamContext failed");
+  // override if you want to force a particular stream
+  nppCtx.hStream = rawStream;
+#else
+  // CUDA 12.9+: helper was removed, we need to build it manually
+  int dev = 0;
+  cudaError_t err = cudaGetDevice(&dev);
+  TORCH_CHECK(err == cudaSuccess, "cudaGetDevice failed");
+  cudaDeviceProp prop{};
+  err = cudaGetDeviceProperties(&prop, dev);
+  TORCH_CHECK(err == cudaSuccess, "cudaGetDeviceProperties failed");
+
+  nppCtx.nCudaDeviceId = dev;
+  nppCtx.nMultiProcessorCount = prop.multiProcessorCount;
+  nppCtx.nMaxThreadsPerMultiProcessor = prop.maxThreadsPerMultiProcessor;
+  nppCtx.nMaxThreadsPerBlock = prop.maxThreadsPerBlock;
+  nppCtx.nSharedMemPerBlock = prop.sharedMemPerBlock;
+  nppCtx.nCudaDevAttrComputeCapabilityMajor = prop.major;
+  nppCtx.nCudaDevAttrComputeCapabilityMinor = prop.minor;
+  nppCtx.nStreamFlags = 0;
+  nppCtx.hStream = rawStream;
+#endif
 
   // Prepare ROI + pointers
-  NppiSize oSizeROI = { width, height };
-  Npp8u* input[2]  = { avFrame->data[0], avFrame->data[1] };
+  NppiSize oSizeROI = {width, height};
+  Npp8u* input[2] = {avFrame->data[0], avFrame->data[1]};
 
   auto start = std::chrono::high_resolution_clock::now();
   NppStatus status;
@@ -281,10 +282,8 @@ void CudaDeviceInterface::convertAVFrameToFrameOutput(
 
   auto end = std::chrono::high_resolution_clock::now();
   auto duration = std::chrono::duration<double, std::micro>(end - start);
-  VLOG(9) << "NPP Conversion of frame h=" << height
-          << " w=" << width
+  VLOG(9) << "NPP Conversion of frame h=" << height << " w=" << width
           << " took: " << duration.count() << "us";
-
 }
 
 // inspired by https://github.com/FFmpeg/FFmpeg/commit/ad67ea9