diff --git a/HeterogeneousCore/CUDAServices/src/CUDAService.cc b/HeterogeneousCore/CUDAServices/src/CUDAService.cc index ee63f1c7219de..2d6e0bfc6fd09 100644 --- a/HeterogeneousCore/CUDAServices/src/CUDAService.cc +++ b/HeterogeneousCore/CUDAServices/src/CUDAService.cc @@ -334,8 +334,8 @@ CUDAService::CUDAService(edm::ParameterSet const& config, edm::ActivityRegistry& log << "cub::CachingDeviceAllocator disabled\n"; } - cudaStreamCache_ = std::make_unique(numberOfDevices_); - cudaEventCache_ = std::make_unique(numberOfDevices_); + cudaStreamCache_ = std::make_unique(lastDevice+1); + cudaEventCache_ = std::make_unique(lastDevice+1); log << "\n";