Create common g_cached_npp_ctxs

NicolasHug · NicolasHug · commit 85d58fbb4c00 · 2025-10-15T13:08:00.000+01:00
diff --git a/src/torchcodec/_core/CUDACommon.cpp b/src/torchcodec/_core/CUDACommon.cpp
@@ -5,6 +5,7 @@
 // LICENSE file in the root directory of this source tree.
 
 #include "src/torchcodec/_core/CUDACommon.h"
+#include "src/torchcodec/_core/Cache.h" // for PerGpuCache
 
 namespace facebook::torchcodec {
 
diff --git a/src/torchcodec/_core/NVDECCache.cpp b/src/torchcodec/_core/NVDECCache.cpp
@@ -22,9 +22,7 @@ namespace facebook::torchcodec {
 
 NVDECCache& NVDECCache::getCache(const torch::Device& device) {
   static NVDECCache cacheInstances[MAX_CUDA_GPUS];
-
-  int deviceIndex = getDeviceIndex(device);
-  return cacheInstances[deviceIndex];
+  return cacheInstances[getDeviceIndex(device)];
 }
 
 UniqueCUvideodecoder NVDECCache::getDecoder(CUVIDEOFORMAT* videoFormat) {

Original file line number	Diff line number	Diff line change
`@@ -22,9 +22,7 @@ namespace facebook::torchcodec {`
`22`	`22`
`23`	`23`	`NVDECCache& NVDECCache::getCache(const torch::Device& device) {`
`24`	`24`	`static NVDECCache cacheInstances[MAX_CUDA_GPUS];`
`25`		`-`
`26`		`- int deviceIndex = getDeviceIndex(device);`
`27`		`- return cacheInstances[deviceIndex];`
	`25`	`+ return cacheInstances[getDeviceIndex(device)];`
`28`	`26`	`}`
`29`	`27`
`30`	`28`	`UniqueCUvideodecoder NVDECCache::getDecoder(CUVIDEOFORMAT* videoFormat) {`