Merge pull request #1026 from krasznaa/AlpakaTemplatingRewrite-main-20250618

Alpaka Templating Rewrite, main branch (2025.06.18.)

Author: krasznaa

device/alpaka/CMakeLists.txt

@@ -51,18 +51,18 @@ traccc_add_alpaka_library( traccc_alpaka alpaka TYPE SHARED
   "src/seeding/seeding_algorithm.cpp"
   "include/traccc/alpaka/seeding/track_params_estimation.hpp"
   "src/seeding/track_params_estimation.cpp"
-  # Finding
-  "include/traccc/alpaka/finding/finding_algorithm.hpp"
-  "src/finding/finding_algorithm.cpp"
-  "src/finding/kernels/make_barcode_sequence.hpp"
-  "src/finding/kernels/apply_interaction.hpp"
-  "src/finding/kernels/fill_sort_keys.hpp"
-  "src/finding/kernels/build_tracks.hpp"
-  "src/finding/kernels/find_tracks.hpp"
-  "src/finding/kernels/propagate_to_next_surface.hpp"
-  # Fitting
-  "include/traccc/alpaka/fitting/fitting_algorithm.hpp"
-  "src/fitting/fitting_algorithm.cpp"
+  # Track finding algorithm(s).
+  "include/traccc/alpaka/finding/combinatorial_kalman_filter_algorithm.hpp"
+  "src/finding/combinatorial_kalman_filter_algorithm.cpp"
+  "src/finding/combinatorial_kalman_filter_algorithm_constant_field_default_detector.cpp"
+  "src/finding/combinatorial_kalman_filter_algorithm_constant_field_telescope_detector.cpp"
+  "src/finding/find_tracks.hpp"
+  # Track fitting algorithm(s).
+  "include/traccc/alpaka/fitting/kalman_fitting_algorithm.hpp"
+  "src/fitting/kalman_fitting_algorithm.cpp"
+  "src/fitting/kalman_fitting_algorithm_constant_field_default_detector.cpp"
+  "src/fitting/kalman_fitting_algorithm_constant_field_telescope_detector.cpp"
+  "src/fitting/fit_tracks.hpp"
 )
 
 # Set up Thrust specifically for the traccc::alpaka library.

device/alpaka/include/traccc/alpaka/finding/combinatorial_kalman_filter_algorithm.hpp

@@ -0,0 +1,110 @@
+/** TRACCC library, part of the ACTS project (R&D line)
+ *
+ * (c) 2025 CERN for the benefit of the ACTS project
+ *
+ * Mozilla Public License Version 2.0
+ */
+
+#pragma once
+
+// Local include(s).
+#include "traccc/alpaka/utils/queue.hpp"
+
+// Project include(s).
+#include "traccc/edm/measurement.hpp"
+#include "traccc/edm/track_candidate_collection.hpp"
+#include "traccc/edm/track_parameters.hpp"
+#include "traccc/finding/finding_config.hpp"
+#include "traccc/geometry/detector.hpp"
+#include "traccc/utils/algorithm.hpp"
+#include "traccc/utils/bfield.hpp"
+#include "traccc/utils/detector_type_utils.hpp"
+#include "traccc/utils/memory_resource.hpp"
+#include "traccc/utils/messaging.hpp"
+
+// VecMem include(s).
+#include <vecmem/utils/copy.hpp>
+
+// System include(s).
+#include <functional>
+
+namespace traccc::alpaka {
+
+/// CKF track finding algorithm
+class combinatorial_kalman_filter_algorithm
+    : public algorithm<edm::track_candidate_collection<default_algebra>::buffer(
+          const default_detector::view&,
+          const covfie::field<traccc::const_bfield_backend_t<
+              default_detector::device::scalar_type>>::view_t&,
+          const measurement_collection_types::const_view&,
+          const bound_track_parameters_collection_types::const_view&)>,
+      public algorithm<edm::track_candidate_collection<default_algebra>::buffer(
+          const telescope_detector::view&,
+          const covfie::field<traccc::const_bfield_backend_t<
+              default_detector::device::scalar_type>>::view_t&,
+          const measurement_collection_types::const_view&,
+          const bound_track_parameters_collection_types::const_view&)>,
+      public messaging {
+
+    public:
+    /// Configuration type
+    using config_type = finding_config;
+    /// Output type
+    using output_type =
+        edm::track_candidate_collection<default_algebra>::buffer;
+
+    /// Constructor with the algorithm's configuration
+    explicit combinatorial_kalman_filter_algorithm(
+        const config_type& config, const traccc::memory_resource& mr,
+        vecmem::copy& copy, queue& q,
+        std::unique_ptr<const Logger> logger = getDummyLogger().clone());
+
+    /// Execute the algorithm
+    ///
+    /// @param det          The (default) detector object
+    /// @param field        The (constant) magnetic field object
+    /// @param measurements All measurements in an event
+    /// @param seeds        All seeds in an event to start the track finding
+    ///                     with
+    ///
+    /// @return A container of the found track candidates
+    ///
+    output_type operator()(
+        const default_detector::view& det,
+        const covfie::field<const_bfield_backend_t<
+            telescope_detector::device::scalar_type>>::view_t& field,
+        const measurement_collection_types::const_view& measurements,
+        const bound_track_parameters_collection_types::const_view& seeds)
+        const override;
+
+    /// Execute the algorithm
+    ///
+    /// @param det          The (telescope) detector object
+    /// @param field        The (constant) magnetic field object
+    /// @param measurements All measurements in an event
+    /// @param seeds        All seeds in an event to start the track finding
+    ///                     with
+    ///
+    /// @return A container of the found track candidates
+    ///
+    output_type operator()(
+        const telescope_detector::view& det,
+        const covfie::field<const_bfield_backend_t<
+            telescope_detector::device::scalar_type>>::view_t& field,
+        const measurement_collection_types::const_view& measurements,
+        const bound_track_parameters_collection_types::const_view& seeds)
+        const override;
+
+    private:
+    /// Algorithm configuration
+    config_type m_config;
+    /// Memory resource used by the algorithm
+    traccc::memory_resource m_mr;
+    /// Copy object used by the algorithm
+    std::reference_wrapper<vecmem::copy> m_copy;
+    /// Alpaka queue
+    std::reference_wrapper<queue> m_queue;
+
+};  // class combinatorial_kalman_filter_algorithm
+
+}  // namespace traccc::alpaka