checkpoint

mconcas · Sep 12, 2024 · 6fecdd1 · 6fecdd1
1 parent 4991e80
commit 6fecdd1
Show file tree

Hide file tree

Showing 4 changed files with 647 additions and 603 deletions.
diff --git a/Common/DCAFitter/GPU/cuda/DCAFitterN.cu b/Common/DCAFitter/GPU/cuda/DCAFitterN.cu
@@ -103,4 +103,46 @@ int process(o2::vertexing::DCAFitterN<2>& fitter,
   return result;
 }
 
+template <int N, class... Tr>
+int process(o2::vertexing::DCAFitterN<2>&,
+            const int nBlocks = 1,
+            const int nThreads = 1,
+            Tr&... args)
+{
+  DCAFitterN<N>* ft_device;
+  std::array<o2::track::TrackParCov*, N> tracks_device;
+  // o2::track::TrackParCov* t1_device;
+  // o2::track::TrackParCov* t2_device;
+  int result, *result_device;
+
+  gpuCheckError(cudaMalloc(reinterpret_cast<void**>(&ft_device), sizeof(o2::vertexing::DCAFitterN<N>)));
+  // gpuCheckError(cudaMalloc(reinterpret_cast<void**>(&t1_device), sizeof(o2::track::TrackParCov)));
+  // gpuCheckError(cudaMalloc(reinterpret_cast<void**>(&t2_device), sizeof(o2::track::TrackParCov)));
+  for (int iT{0}; iT < N; ++iT) {
+    gpuCheckError(cudaMalloc(reinterpret_cast<void**>(&(tracks_device[iT])), sizeof(o2::track::TrackParCov)));
+  }
+  gpuCheckError(cudaMalloc(reinterpret_cast<void**>(&result_device), sizeof(int)));
+
+  gpuCheckError(cudaMemcpy(ft_device, &fitter, sizeof(o2::vertexing::DCAFitterN<2>), cudaMemcpyHostToDevice));
+  gpuCheckError(cudaMemcpy(t1_device, &track1, sizeof(o2::track::TrackParCov), cudaMemcpyHostToDevice));
+  gpuCheckError(cudaMemcpy(t2_device, &track2, sizeof(o2::track::TrackParCov), cudaMemcpyHostToDevice));
+
+  kernel::processKernel<<<nBlocks, nThreads>>>(ft_device, t1_device, t2_device, result_device);
+
+  gpuCheckError(cudaPeekAtLastError());
+  gpuCheckError(cudaDeviceSynchronize());
+
+  gpuCheckError(cudaMemcpy(&result, result_device, sizeof(int), cudaMemcpyDeviceToHost));
+  gpuCheckError(cudaMemcpy(&fitter, ft_device, sizeof(o2::vertexing::DCAFitterN<2>), cudaMemcpyDeviceToHost));
+  gpuCheckError(cudaMemcpy(&track1, t1_device, sizeof(o2::track::TrackParCov), cudaMemcpyDeviceToHost));
+  gpuCheckError(cudaMemcpy(&track2, t2_device, sizeof(o2::track::TrackParCov), cudaMemcpyDeviceToHost));
+  gpuCheckError(cudaFree(ft_device));
+  gpuCheckError(cudaFree(t1_device));
+  gpuCheckError(cudaFree(t2_device));
+
+  gpuCheckError(cudaFree(result_device));
+
+  return result;
+}
+
 } // namespace o2::vertexing::device