PixelVertexFinding/plugins/PixelVertexWorkSpaceSoAHost.h

0001 #ifndef RecoVertex_PixelVertexFinding_plugins_PixelVertexWorkSpaceSoAHost_h
0002 #define RecoVertex_PixelVertexFinding_plugins_PixelVertexWorkSpaceSoAHost_h
0003
0004 #include "CUDADataFormats/Common/interface/PortableHostCollection.h"
0005 #include "CUDADataFormats/Vertex/interface/ZVertexUtilities.h"
0006 #include "RecoVertex/PixelVertexFinding/plugins/PixelVertexWorkSpaceUtilities.h"
0007
0008 template <int32_t S>
0009 class PixelVertexWorkSpaceSoAHost : public cms::cuda::PortableHostCollection<PixelVertexWSSoALayout<>> {
0010 public:
0011   explicit PixelVertexWorkSpaceSoAHost() : PortableHostCollection<PixelVertexWSSoALayout<>>(S) {}
0012
0013   // Constructor which specifies the SoA size and CUDA stream
0014   explicit PixelVertexWorkSpaceSoAHost(cudaStream_t stream)
0015       : PortableHostCollection<PixelVertexWSSoALayout<>>(S, stream) {}
0016 };
0017
0018 namespace gpuVertexFinder {
0019   namespace workSpace {
0020     using PixelVertexWorkSpaceSoAHost = PixelVertexWorkSpaceSoAHost<zVertex::utilities::MAXTRACKS>;
0021   }
0022 }  // namespace gpuVertexFinder
0023 #endif