-
Notifications
You must be signed in to change notification settings - Fork 19
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
8e690ad
commit 3c71893
Showing
4 changed files
with
195 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,163 @@ | ||
| /** Detray library, part of the ACTS project (R&D line) | ||
| * | ||
| * (c) 2024 CERN for the benefit of the ACTS project | ||
| * | ||
| * Mozilla Public License Version 2.0 | ||
| */ | ||
|
|
||
| // Project include(s) | ||
| #include "detray/detectors/bfield.hpp" | ||
| #include "detray/navigation/navigator.hpp" | ||
| #include "detray/propagator/actor_chain.hpp" | ||
| #include "detray/propagator/actors/aborters.hpp" | ||
| #include "detray/propagator/actors/parameter_resetter.hpp" | ||
| #include "detray/propagator/actors/parameter_transporter.hpp" | ||
| #include "detray/propagator/actors/pointwise_material_interactor.hpp" | ||
| #include "detray/propagator/rk_stepper.hpp" | ||
| #include "detray/tracks/tracks.hpp" | ||
|
|
||
| // Detray IO include(s) | ||
| #include "detray/io/frontend/detector_reader.hpp" | ||
|
|
||
| // Detray benchmark include(s) | ||
| #include "detray/benchmarks/device/cuda/propagation_benchmark.hpp" | ||
|
|
||
| // Detray test include(s). | ||
| #include "detray/test/utils/simulation/event_generator/track_generators.hpp" | ||
| #include "detray/test/utils/types.hpp" | ||
|
|
||
| // Detray test include(s) | ||
| #include "detray/options/detector_io_options.hpp" | ||
| #include "detray/options/parse_options.hpp" | ||
| #include "detray/options/propagation_options.hpp" | ||
| #include "detray/options/track_generator_options.hpp" | ||
|
|
||
| // Vecmem include(s) | ||
| #include <vecmem/memory/host_memory_resource.hpp> | ||
|
|
||
| // System include(s) | ||
| #include <algorithm> | ||
| #include <string> | ||
|
|
||
| namespace po = boost::program_options; | ||
|
|
||
| using namespace detray; | ||
|
|
||
| int main(int argc, char** argv) { | ||
|
|
||
| // Use the most general type to be able to read in all detector files | ||
| using detector_t = detray::detector<test::default_metadata>; | ||
| using test_algebra = typename detector_t::algebra_type; | ||
| using scalar = dscalar<test_algebra>; | ||
| using vector3 = dvector3D<test_algebra>; | ||
| using free_track_parameters_t = free_track_parameters<test_algebra>; | ||
| using uniform_gen_t = | ||
| detail::random_numbers<scalar, std::uniform_real_distribution<scalar>>; | ||
| using track_generator_t = | ||
| random_track_generator<free_track_parameters_t, uniform_gen_t>; | ||
|
|
||
| using field_bknd_t = bfield::const_bknd_t<scalar>; | ||
|
|
||
| // Host and device memory resources | ||
| vecmem::host_memory_resource host_mr; | ||
| vecmem::cuda::device_memory_resource dev_mr; | ||
|
|
||
| // Constant magnetic field | ||
| vector3 B{0.f, 0.f, 2.f * unit<scalar>::T}; | ||
|
|
||
| // Number of tracks in the different benchmark cases | ||
| std::vector<int> n_tracks{8 * 8, 16 * 16, 32 * 32, 64 * 64, | ||
| 128 * 128, 256 * 256, 512 * 512}; | ||
|
|
||
| // | ||
| // Configuration | ||
| // | ||
|
|
||
| // Specific options for this test | ||
| po::options_description desc("\ndetray propagation benchmark options"); | ||
|
|
||
| std::vector<dindex> window; | ||
| desc.add_options()("context", po::value<dindex>(), | ||
| "Number of the geometry context")( | ||
| "sort_tracks", "Sort track samples by theta angle"); | ||
|
|
||
| // Configs to be filled | ||
| detray::io::detector_reader_config reader_cfg{}; | ||
| track_generator_t::configuration trk_cfg{}; | ||
| propagation::config prop_cfg{}; | ||
| detray::benchmarks::benchmark_base::configuration bench_cfg{}; | ||
|
|
||
| // Read options from commandline | ||
| po::variables_map vm = detray::options::parse_options( | ||
| desc, argc, argv, reader_cfg, trk_cfg, prop_cfg); | ||
|
|
||
| // General options | ||
| bool do_sort{(vm.count("sort_tracks") != 0)}; | ||
|
|
||
| // The geometry context to be displayed | ||
| detector_t::geometry_context gctx; | ||
| if (vm.count("context")) { | ||
| gctx = detector_t::geometry_context{vm["context"].as<dindex>()}; | ||
| } | ||
|
|
||
| // | ||
| // Prepare data | ||
| // | ||
|
|
||
| // Read the detector geometry | ||
| reader_cfg.do_check(true); | ||
|
|
||
| const auto [det, names] = | ||
| detray::io::read_detector<detector_t>(host_mr, reader_cfg); | ||
| const std::string& det_name = det.name(names); | ||
|
|
||
| // Generate the track samples | ||
| auto track_samples = | ||
| detray::benchmarks::generate_track_samples<track_generator_t>( | ||
| &host_mr, n_tracks, trk_cfg, do_sort); | ||
|
|
||
| // Create a constant b-field | ||
| auto bfield = bfield::create_const_field<scalar>(B); | ||
|
|
||
| // Build actor states | ||
| dtuple<> empty_state{}; | ||
|
|
||
| parameter_transporter<test_algebra>::state transporter_state{}; | ||
| pointwise_material_interactor<test_algebra>::state interactor_state{}; | ||
| parameter_resetter<test_algebra>::state resetter_state{}; | ||
|
|
||
| auto actor_states = detail::make_tuple<dtuple>( | ||
| transporter_state, interactor_state, resetter_state); | ||
|
|
||
| // | ||
| // Register benchmarks | ||
| // | ||
|
|
||
| // Number of warmup tracks | ||
| const int n_max_tracks{*std::ranges::max_element(n_tracks)}; | ||
| bench_cfg.n_warmup( | ||
| static_cast<int>(std::ceil(0.1f * static_cast<float>(n_max_tracks)))); | ||
|
|
||
| if (prop_cfg.stepping.do_covariance_transport) { | ||
| detray::benchmarks::register_benchmark< | ||
| detray::benchmarks::cuda_propagation_bm, | ||
| detray::benchmarks::cuda_propagator_type< | ||
| test::default_metadata, field_bknd_t, | ||
| detray::benchmarks::default_chain>>( | ||
| det_name + "_W_COV_TRANSPORT", bench_cfg, prop_cfg, det, bfield, | ||
| &actor_states, track_samples, n_tracks, &dev_mr); | ||
| } else { | ||
| detray::benchmarks::register_benchmark< | ||
| detray::benchmarks::cuda_propagation_bm, | ||
| detray::benchmarks::cuda_propagator_type< | ||
| test::default_metadata, field_bknd_t, | ||
| detray::benchmarks::empty_chain>>( | ||
| det_name, bench_cfg, prop_cfg, det, bfield, &empty_state, | ||
| track_samples, n_tracks, &dev_mr); | ||
| } | ||
|
|
||
| // Run benchmarks | ||
| ::benchmark::Initialize(&argc, argv); | ||
| ::benchmark::RunSpecifiedBenchmarks(); | ||
| ::benchmark::Shutdown(); | ||
| } |