PixelVertexFinding/plugins/PixelVertexWorkSpaceSoAHost.h

0001 #ifndef RecoPixelVertexing_PixelVertexFinding_PixelVertexWorkSpaceSoAHost_h
0002 #define RecoPixelVertexing_PixelVertexFinding_PixelVertexWorkSpaceSoAHost_h
0003
0004 #include "CUDADataFormats/Common/interface/PortableHostCollection.h"
0005 #include "CUDADataFormats/Vertex/interface/ZVertexUtilities.h"
0006 #include "RecoPixelVertexing/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   // Constructor which specifies the SoA size and CUDA stream
0013   explicit PixelVertexWorkSpaceSoAHost(cudaStream_t stream)
0014       : PortableHostCollection<PixelVertexWSSoALayout<>>(S, stream) {}
0015 };
0016
0017 namespace gpuVertexFinder {
0018   namespace workSpace {
0019     using PixelVertexWorkSpaceSoAHost = PixelVertexWorkSpaceSoAHost<zVertex::utilities::MAXTRACKS>;
0020   }
0021 }  // namespace gpuVertexFinder
0022 #endif