device/cuda/propagator_cuda_kernel.hpp

0001 // This file is part of the ACTS project.
0002 //
0003 // Copyright (C) 2016 CERN for the benefit of the ACTS project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
0008
0009 #pragma once
0010
0011 // Project include(s)
0012 #include "detray/core/detector.hpp"
0013 #include "detray/detectors/toy_metadata.hpp"
0014
0015 // Detray test include(s)
0016 #include "detray/test/device/cuda/bfield.hpp"
0017 #include "detray/test/device/propagator_test.hpp"
0018
0019 // Vecmem include(s)
0020 #include <vecmem/memory/memory_resource.hpp>
0021 #include <vecmem/utils/cuda/copy.hpp>
0022
0023 namespace detray {
0024
0025 using scalar = test::scalar;
0026
0027 /// Launch the propagation test kernel
0028 template <typename bfield_bknd_t, typename detector_t>
0029 void propagator_test(
0030     typename detector_t::view_type, const propagation::config &,
0031     covfie::field_view<bfield_bknd_t>, vecmem::data::vector_view<test_track> &,
0032     vecmem::data::jagged_vector_view<detail::step_data<test_algebra>> &);
0033
0034 /// Test function for propagator on the device
0035 template <typename bfield_bknd_t, typename detector_t>
0036 inline auto run_propagation_device(
0037     vecmem::memory_resource *mr, const propagation::config &cfg,
0038     typename detector_t::view_type det_view,
0039     covfie::field_view<bfield_bknd_t> field_data, dvector<test_track> &tracks,
0040     const vecmem::jagged_vector<detail::step_data<test_algebra>> &host_steps)
0041     -> vecmem::jagged_vector<detail::step_data<test_algebra>> {
0042   // Helper object for performing memory copies.
0043   vecmem::copy copy;
0044
0045   // Get tracks data
0046   auto tracks_data = vecmem::get_data(tracks);
0047
0048   // Create vector buffer for track recording
0049   std::vector<std::size_t> sizes(tracks.size(), 0);
0050   std::vector<std::size_t> capacities;
0051   for (auto &st : host_steps) {
0052     // Add a few more elements for security (in case the device finds more
0053     // surfaces)
0054     capacities.push_back(st.size() + 10u);
0055   }
0056
0057   vecmem::data::jagged_vector_buffer<detail::step_data<test_algebra>>
0058       steps_buffer(capacities, *mr, nullptr,
0059                    vecmem::data::buffer_type::resizable);
0060
0061   copy.setup(steps_buffer)->wait();
0062
0063   // Run the propagator test for GPU device
0064   propagator_test<bfield_bknd_t, detector_t>(det_view, cfg, field_data,
0065                                              tracks_data, steps_buffer);
0066
0067   vecmem::jagged_vector<detail::step_data<test_algebra>> steps(mr);
0068
0069   copy(steps_buffer, steps)->wait();
0070
0071   return steps;
0072 }
0073
0074 /// Test chain for the propagator
0075 template <typename device_bfield_bknd_t, typename host_bfield_bknd_t,
0076           typename detector_t>
0077 inline auto run_propagation_test(vecmem::memory_resource *mr, detector_t &det,
0078                                  const propagator_test_config &cfg,
0079                                  typename detector_t::view_type det_view,
0080                                  covfie::field<host_bfield_bknd_t> &&field) {
0081   // Create the vector of initial track parameterizations
0082   auto tracks_host = generate_tracks<generator_t>(mr, cfg.track_generator);
0083   vecmem::vector<test_track> tracks_device(tracks_host, mr);
0084
0085   // Host propagation
0086   auto host_steps =
0087       run_propagation_host(mr, det, cfg.propagation, field, tracks_host);
0088
0089   // Device propagation (device backend specific implementation)
0090   covfie::field<device_bfield_bknd_t> device_field(field);
0091   auto device_steps = run_propagation_device<device_bfield_bknd_t, detector_t>(
0092       mr, cfg.propagation, det_view, device_field, tracks_device, host_steps);
0093
0094   // Check the results
0095   compare_propagation_results(host_steps, device_steps);
0096 }
0097
0098 }  // namespace detray