#include <move_kernel.hpp>

Collaboration diagram for Simulation::KernelInline::MoveFunctor:

Public Types
using	TeamPolicy = Kokkos::TeamPolicy<ComputeSpace>
using	TeamMember = TeamPolicy::member_type

Public Member Functions
	MoveFunctor ()=default
	MoveFunctor (std::size_t p_team_move, std::size_t p_team_leave, MC::ParticlePositions p, MC::ParticleStatus _status, MC::DomainState< ComputeSpace > m, MC::pool_type _random_pool, MC::EventContainer _events, ProbeAutogeneratedBuffer _probes, MC::ParticleAges _ages)
	MoveFunctor (std::size_t p_team_move, std::size_t p_team_leave, double _d_t, MC::ParticlePositions p, MC::ParticleStatus _status, std::size_t n_p, MC::DomainState< ComputeSpace > m, MC::pool_type _random_pool, MC::EventContainer _events, ProbeAutogeneratedBuffer _probes, MC::ParticleAges _ages, bool b_move, bool b_leave)
void	update (double _d_t, std::size_t n_p, MC::DomainState< ComputeSpace > move_i, MC::ParticlePositions _positions, MC::ParticleStatus _status, MC::ParticleAges _ages, bool b_move, bool b_leave)
bool	need_launch () const
KOKKOS_INLINE_FUNCTION void	operator() (TagMove, const Kokkos::TeamPolicy< ComputeSpace >::member_type &team, MoveLeaveTally &local_tally) const
KOKKOS_INLINE_FUNCTION void	operator() (TagLeave _tag, const Kokkos::TeamPolicy< ComputeSpace >::member_type &team_handle, std::size_t &local_dead_count) const
KOKKOS_INLINE_FUNCTION void	operator() (TagMoveLeave, const Kokkos::TeamPolicy< ComputeSpace >::member_type &team, MoveLeaveTally &local_tally) const
bool	do_move () const
bool	do_leave () const
KOKKOS_INLINE_FUNCTION void	operator() (TagLeaveB0D _tag, const Kokkos::TeamPolicy< ComputeSpace >::member_type &team_handle, MoveLeaveTally &local_tally) const
template<typename GenType>
KOKKOS_FORCEINLINE_FUNCTION std::size_t	next_compartment (const std::size_t i_current_compartment, const MC::MoveProbabilityView< ComputeSpace, true > &move_probability, GenType &gen, bool &moved) const
template<typename ExecSpace, typename GenType>
KOKKOS_FORCEINLINE_FUNCTION void	handle_exit_at (const std::size_t idx, const Kokkos::View< const MC::LeavingFlow *, ExecSpace > &leaving_flow, const std::size_t position, GenType &gen, std::size_t &dead_count) const
template<typename ExecSpace>
KOKKOS_FORCEINLINE_FUNCTION std::size_t	handle_exit (const std::size_t idx, const Kokkos::View< const MC::LeavingFlow *, ExecSpace > &leaving_flow, std::size_t &dead_count) const
KOKKOS_INLINE_FUNCTION void	perform_exit (const bool proba, const std::size_t idx, std::size_t &dead_count) const

Public Attributes
double	d_t {}
MC::ParticlePositions	positions
std::size_t	n_particles {}
MC::DomainState< ComputeSpace, true >	move
MC::pool_type	random_pool
MC::ParticleStatus	status
MC::EventContainer	events
ProbeAutogeneratedBuffer	probes
MC::ParticleAges	ages
std::size_t	m_p_team_leave {}
std::size_t	m_p_team_move {}
bool	m_enable_move {}
bool	m_enable_leave {}

Member Typedef Documentation

◆ TeamMember

using Simulation::KernelInline::MoveFunctor::TeamMember = TeamPolicy::member_type

◆ TeamPolicy

using Simulation::KernelInline::MoveFunctor::TeamPolicy = Kokkos::TeamPolicy<ComputeSpace>

Constructor & Destructor Documentation

◆ MoveFunctor() [1/3]

Simulation::KernelInline::MoveFunctor::MoveFunctor ( )

default

Here is the caller graph for this function:

◆ MoveFunctor() [2/3]

Simulation::KernelInline::MoveFunctor::MoveFunctor	(	std::size_t	p_team_move,
		std::size_t	p_team_leave,
		MC::ParticlePositions	p,
		MC::ParticleStatus	_status,
		MC::DomainState< ComputeSpace >	m,
		MC::pool_type	_random_pool,
		MC::EventContainer	_events,
		ProbeAutogeneratedBuffer	_probes,
		MC::ParticleAges	_ages )

inline

Here is the call graph for this function:

◆ MoveFunctor() [3/3]

Simulation::KernelInline::MoveFunctor::MoveFunctor	(	std::size_t	p_team_move,
		std::size_t	p_team_leave,
		double	_d_t,
		MC::ParticlePositions	p,
		MC::ParticleStatus	_status,
		std::size_t	n_p,
		MC::DomainState< ComputeSpace >	m,
		MC::pool_type	_random_pool,
		MC::EventContainer	_events,
		ProbeAutogeneratedBuffer	_probes,
		MC::ParticleAges	_ages,
		bool	b_move,
		bool	b_leave )

inline

Member Function Documentation

◆ do_leave()

bool Simulation::KernelInline::MoveFunctor::do_leave ( ) const

inline

◆ do_move()

bool Simulation::KernelInline::MoveFunctor::do_move ( ) const

inline

◆ handle_exit()

template<typename ExecSpace>

KOKKOS_FORCEINLINE_FUNCTION std::size_t Simulation::KernelInline::MoveFunctor::handle_exit	(	const std::size_t	idx,
		const Kokkos::View< const MC::LeavingFlow *, ExecSpace > &	leaving_flow,
		std::size_t &	dead_count ) const

inline

Here is the call graph for this function:

Here is the caller graph for this function:

◆ handle_exit_at()

template<typename ExecSpace, typename GenType>

KOKKOS_FORCEINLINE_FUNCTION void Simulation::KernelInline::MoveFunctor::handle_exit_at	(	const std::size_t	idx,
		const Kokkos::View< const MC::LeavingFlow *, ExecSpace > &	leaving_flow,
		const std::size_t	position,
		GenType &	gen,
		std::size_t &	dead_count ) const

inline

Exit test for a particle whose position the caller already holds, using a generator the caller already owns. Avoids both the re-read of positions and the per-particle get_state/free_state round trip.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ need_launch()

bool Simulation::KernelInline::MoveFunctor::need_launch ( ) const

inlinenodiscard

◆ next_compartment()

template<typename GenType>

KOKKOS_FORCEINLINE_FUNCTION std::size_t Simulation::KernelInline::MoveFunctor::next_compartment	(	const std::size_t	i_current_compartment,
		const MC::MoveProbabilityView< ComputeSpace, true > &	move_probability,
		GenType &	gen,
		bool &	moved ) const

inline

Pure form of the move: no access to positions, so the caller keeps the compartment in a register.

Returns: the compartment the particle ends up in; moved is set when it actually changed compartment.

Here is the call graph for this function:

Here is the caller graph for this function:

◆ operator()() [1/4]

KOKKOS_INLINE_FUNCTION void Simulation::KernelInline::MoveFunctor::operator()	(	TagLeave	_tag,
		const Kokkos::TeamPolicy< ComputeSpace >::member_type &	team_handle,
		std::size_t &	local_dead_count ) const

inline

Here is the call graph for this function:

◆ operator()() [2/4]

KOKKOS_INLINE_FUNCTION void Simulation::KernelInline::MoveFunctor::operator()	(	TagLeaveB0D	_tag,
		const Kokkos::TeamPolicy< ComputeSpace >::member_type &	team_handle,
		MoveLeaveTally &	local_tally ) const

inline

Here is the call graph for this function:

◆ operator()() [3/4]

KOKKOS_INLINE_FUNCTION void Simulation::KernelInline::MoveFunctor::operator()	(	TagMove	,
		const Kokkos::TeamPolicy< ComputeSpace >::member_type &	team,
		MoveLeaveTally &	local_tally ) const

inline

Here is the call graph for this function:

◆ operator()() [4/4]

KOKKOS_INLINE_FUNCTION void Simulation::KernelInline::MoveFunctor::operator()	(	TagMoveLeave	,
		const Kokkos::TeamPolicy< ComputeSpace >::member_type &	team,
		MoveLeaveTally &	local_tally ) const

inline

Fused move + leave. Each thread holds one generator for its whole strided chunk, the same way TagLeaveB0D does.

Here is the call graph for this function:

◆ perform_exit()

KOKKOS_INLINE_FUNCTION void Simulation::KernelInline::MoveFunctor::perform_exit	(	const bool	proba,
		const std::size_t	idx,
		std::size_t &	dead_count ) const

inline

Here is the caller graph for this function:

◆ update()

void Simulation::KernelInline::MoveFunctor::update	(	double	_d_t,
		std::size_t	n_p,
		MC::DomainState< ComputeSpace >	move_i,
		MC::ParticlePositions	_positions,
		MC::ParticleStatus	_status,
		MC::ParticleAges	_ages,
		bool	b_move,
		bool	b_leave )

inline

Member Data Documentation

◆ ages

MC::ParticleAges Simulation::KernelInline::MoveFunctor::ages

◆ d_t

double Simulation::KernelInline::MoveFunctor::d_t {}

◆ events

MC::EventContainer Simulation::KernelInline::MoveFunctor::events

◆ m_enable_leave

bool Simulation::KernelInline::MoveFunctor::m_enable_leave {}

◆ m_enable_move

bool Simulation::KernelInline::MoveFunctor::m_enable_move {}

◆ m_p_team_leave

std::size_t Simulation::KernelInline::MoveFunctor::m_p_team_leave {}

◆ m_p_team_move

std::size_t Simulation::KernelInline::MoveFunctor::m_p_team_move {}

◆ move

MC::DomainState<ComputeSpace, true> Simulation::KernelInline::MoveFunctor::move

◆ n_particles

std::size_t Simulation::KernelInline::MoveFunctor::n_particles {}

◆ positions

MC::ParticlePositions Simulation::KernelInline::MoveFunctor::positions

◆ probes

ProbeAutogeneratedBuffer Simulation::KernelInline::MoveFunctor::probes

◆ random_pool

MC::pool_type Simulation::KernelInline::MoveFunctor::random_pool

◆ status

MC::ParticleStatus Simulation::KernelInline::MoveFunctor::status

The documentation for this struct was generated from the following file:

apps/libs/simulation/public/simulation/kernels/move_kernel.hpp

Public Types

Public Member Functions

Public Attributes

Member Typedef Documentation

◆ TeamMember

◆ TeamPolicy

Constructor & Destructor Documentation

◆ MoveFunctor() [1/3]

◆ MoveFunctor() [2/3]

◆ MoveFunctor() [3/3]

Member Function Documentation

◆ do_leave()

◆ do_move()

◆ handle_exit()

◆ handle_exit_at()

◆ need_launch()

◆ next_compartment()

◆ operator()() [1/4]

◆ operator()() [2/4]

◆ operator()() [3/4]

◆ operator()() [4/4]

◆ perform_exit()

◆ update()

Member Data Documentation

◆ ages

◆ d_t

◆ events

◆ m_enable_leave

◆ m_enable_move

◆ m_p_team_leave

◆ m_p_team_move

◆ move

◆ n_particles

◆ positions

◆ probes

◆ random_pool

◆ status