-
Notifications
You must be signed in to change notification settings - Fork 218
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
ChooseInstantNonReversibleTransition kernel
ci: picongpu
- Loading branch information
1 parent
ab9408a
commit 6261bd5
Showing
14 changed files
with
1,182 additions
and
51 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
44 changes: 44 additions & 0 deletions
44
include/picongpu/particles/atomicPhysics/InstantTransitionRateLimit.hpp
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,44 @@ | ||
| /* Copyright 2024 Brian Marre | ||
| * | ||
| * This file is part of PIConGPU. | ||
| * | ||
| * PIConGPU is free software: you can redistribute it and/or modify | ||
| * it under the terms of the GNU General Public License as published by | ||
| * the Free Software Foundation, either version 3 of the License, or | ||
| * (at your option) any later version. | ||
| * | ||
| * PIConGPU is distributed in the hope that it will be useful, | ||
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| * GNU General Public License for more details. | ||
| * | ||
| * You should have received a copy of the GNU General Public License | ||
| * along with PIConGPU. | ||
| * If not, see <http://www.gnu.org/licenses/>. | ||
| */ | ||
|
|
||
| #pragma once | ||
|
|
||
| // need unit system and normalization | ||
| #include "picongpu/defines.hpp" | ||
| #include "picongpu/particles/atomicPhysics/param.hpp" | ||
|
|
||
| namespace picongpu::particles::atomicPhysics | ||
| { | ||
| struct InstantTransitionRateLimit | ||
| { | ||
| /** get limit of total state loss rate for inclusion n the time dependent rate equation solver | ||
| * | ||
| * @tparam T_ReturnType type and precision to use in the result | ||
| */ | ||
| template<typename T_ReturnType> | ||
| static constexpr T_ReturnType get() | ||
| { | ||
| using picongpu::atomicPhysics::RateSolverParam; | ||
|
|
||
| // unit: unitless * unitless / unit_time = 1/unit_time | ||
| return static_cast<T_ReturnType>( | ||
| RateSolverParam::timeStepAlpha * float_X(RateSolverParam::maximumNumberSubStepsPerPICTimeStep)) | ||
| / picongpu::sim.pic.getDt<T_ReturnType>(); | ||
| } | ||
| } // namespace picongpu::particles::atomicPhysics |
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
123 changes: 123 additions & 0 deletions
123
include/picongpu/particles/atomicPhysics/kernel/CheckForFieldEnergyOverSubscription.kernel
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,123 @@ | ||
| /* Copyright 2024 Brian Marre | ||
| * | ||
| * This file is part of PIConGPU. | ||
| * | ||
| * PIConGPU is free software: you can redistribute it and/or modify | ||
| * it under the terms of the GNU General Public License as published by | ||
| * the Free Software Foundation, either version 3 of the License, or | ||
| * (at your option) any later version. | ||
| * | ||
| * PIConGPU is distributed in the hope that it will be useful, | ||
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| * GNU General Public License for more details. | ||
| * | ||
| * You should have received a copy of the GNU General Public License | ||
| * along with PIConGPU. | ||
| * If not, see <http://www.gnu.org/licenses/>. | ||
| */ | ||
|
|
||
| #pragma once | ||
|
|
||
| #include "picongpu/defines.hpp" | ||
| #include "picongpu/particles/atomicPhysics/ConvertEnum.hpp" | ||
| #include "picongpu/particles/atomicPhysics/KernelIndexation.hpp" | ||
| #include "picongpu/particles/atomicPhysics/kernel/CalculateRejectionProbability.hpp" | ||
| #include "picongpu/particles/atomicPhysics/localHelperFields/RejectionProbabilityCache_Cell.hpp" | ||
|
|
||
| #include <pmacc/lockstep/ForEach.hpp> | ||
|
|
||
| #include <cstdint> | ||
|
|
||
| namespace picongpu::particles::atomicPhysics::kernel | ||
| { | ||
| /** check all cells for overSubscription | ||
| * | ||
| * checks each superCell cell for fieldEnergyUse > fieldEnergy, | ||
| * if yes mark cell as oversubscribed and stores rejection probability in rejection probability cache | ||
| * | ||
| * @tparam T_numberAtomicPhysicsIonSpecies specialization template parameter used to prevent compilation of all | ||
| * atomicPhysics kernels if no atomic physics species is present. | ||
| */ | ||
| template<uint32_t T_numberAtomicPhysicsIonSpecies> | ||
| struct CheckForFieldEnergyOverSubscriptionKernel | ||
| { | ||
| /** call operator | ||
| * | ||
| * called by CheckForOverSubscription atomic physics sub-stage | ||
| * | ||
| * @param worker object containing the device and block information, passed by PMACC_KERNEL call | ||
| * @param areMapping mapping of blockIndex to block superCell index | ||
| * @param rngFactory factory for uniformly distributed random number generator | ||
| * @param eFieldBox deviceDataBox giving access to eField values for all local superCells | ||
| * @param fieldEnergyUseCacheBox deviceDataBox giving access to the field energy use cache for each local | ||
| * superCell | ||
| * @param sharedResourcesOverSubscribedBox deviceDataBox giving access to the local shared resources over | ||
| * subscription switch for each local superCell | ||
| * @param rejectionProbabilityCacheCellBox deviceDataBox giving access to localRejectionProbabilityCache for | ||
| * all local superCells | ||
| */ | ||
| template< | ||
| typename T_Worker, | ||
| typename T_AreaMapping, | ||
| typename T_TimeRemainingBox, | ||
| typename T_EFieldBox, | ||
| typename T_FieldEnergyUseCacheBox, | ||
| typename T_SharedRessourcesOverSubscribedBox, | ||
| typename T_RejectionProbabilityCacheCellDataBox> | ||
| HDINLINE void operator()( | ||
| T_Worker const& worker, | ||
| T_AreaMapping const areaMapping, | ||
| T_TimeRemainingBox const timeRemainingBox, | ||
| T_EFieldBox const eFieldBox, | ||
| T_FieldEnergyUseCacheBox const fieldEnergyUseCacheBox, | ||
| T_SharedRessourcesOverSubscribedBox sharedResourcesOverSubscribedBox, | ||
| T_RejectionProbabilityCacheCellDataBox rejectionProbabilityCacheCellBox) const | ||
| { | ||
| auto const superCellIdx = KernelIndexation::getSuperCellIndex(worker, areaMapping); | ||
| auto const superCellFieldIdx | ||
| = KernelIndexation::getSuperCellFieldIndexFromSuperCellIndex(areaMapping, superCellIdx); | ||
|
|
||
| auto const timeRemaining = timeRemainingBox(superCellFieldIdx); | ||
| if(timeRemaining <= 0._X) | ||
| return; | ||
|
|
||
| using FieldEnergyCache = typename T_FieldEnergyUseCacheBox::ValueType; | ||
| particles::atomicPhysics::localHelperFields::RejectionProbabilityCache_Cell<FieldEnergyCache::numberCells>& | ||
| rejectionProbabilityCacheCell | ||
| = rejectionProbabilityCacheCellBox(superCellFieldIdx); | ||
|
|
||
| /// @todo split rejectionProbabilityCache into cell and bin to avoid loading everything?, Brian Marre, 2024 | ||
| FieldEnergyCache const& eFieldEnergyUseCache = fieldEnergyUseCacheBox(superCellFieldIdx); | ||
|
|
||
| bool sharedResourcesOverSubscribed = false; | ||
| DataSpace<picongpu::simDim> const superCellCellOffset = superCellIdx * picongpu::SuperCellSize::toRT(); | ||
|
|
||
| auto forEachCell = pmacc::lockstep::makeForEach<::numberCells, T_Worker>(worker); | ||
| forEachCell( | ||
| [&worker, | ||
| &superCellCellOffset, | ||
| &eFieldBox, | ||
| &eFieldEnergyUseCache, | ||
| &rejectionProbabilityCacheCell, | ||
| &sharedResourcesOverSubscribed](uint32_t const linearCellIndex) | ||
| { | ||
| CalculateRejectionProbability::ofCell( | ||
| linearCellIndex, | ||
| superCellCellOffset, | ||
| eFieldBox, | ||
| eFieldEnergyUseCache, | ||
| rejectionProbabilityCacheCell, | ||
| sharedResourcesOverSubscribed); | ||
| }); | ||
|
|
||
| uint32_t& flagField = sharedResourcesOverSubscribedBox(superCellFieldIdx); | ||
| // write out flag setting to device memory | ||
| alpaka::atomicOr( | ||
| worker.getAcc(), | ||
| &flagField, | ||
| u32(sharedResourcesOverSubscribed), | ||
| ::alpaka::hierarchy::Threads{}); | ||
| } | ||
| }; | ||
| } // namespace picongpu::particles::atomicPhysics::kernel |
Oops, something went wrong.