#include <load_balance.hpp>

Collaboration diagram for LoadBalance:

Public Types
enum class	WeightingAlgorithm { SingleRegionBalance =0 , ParticleBalance , Fortran , Default }

enum class	ConstraintOption { ParticleCount =0 }

enum class	ReweightOption { Always =0 , IfBetter }

Public Member Functions
	LoadBalance ()

	LoadBalance (NLReader::NamelistReader &nlr, const DomainDecomposition< DeviceType > &pol_decomp, bool sync_planes=true)

void	rebalance (const Simulation< DeviceType > &sml, const Grid< DeviceType > &grid, const MagneticField< DeviceType > &magnetic_field, const VelocityGrid &vgrid, Plasma &plasma, DomainDecomposition< DeviceType > &pol_decomp, int istep, int gstep, ReweightOption reweight_option, WeightingAlgorithm weighting_algorithm=WeightingAlgorithm::Default)

void	rebalance (DomainDecomposition< DeviceType > &pol_decomp, const View< double , CLayout, HostType > &constraint, const std::vector< double > &timings, double &load_imbalance, View< double , HostType > &model_belief)

Private Member Functions
double	get_even_division (const View< double *, HostType > &input, int n) const

bool	greedily_fill_partition (const View< double , HostType > &weight, const View< double , HostType > &constraint1, double target_weight_per_rank)

double	get_largest_predicted_time (const View< int , CLayout, HostType > &partition, const View< double , HostType > &weight) const

void	one_weight_balance (const View< double , HostType > &weight, const View< double , CLayout, HostType > constraint1)

bool	recommend_proposed_partition ()

void	print_new_partition ()

void	update_model (const View< int *, CLayout, HostType > &current_partition)

void	update_model (const View< int *, CLayout, HostType > &current_partition, const std::vector< double > &manual_times)

void	initialize_model ()

bool	model_is_initialized ()

void	propose_new_partition (const Kokkos::View< double *, Kokkos::LayoutRight, HostType > &ptl_count, WeightingAlgorithm weighting_algorithm)

void	set_new_partition (const Simulation< DeviceType > &sml, const Grid< DeviceType > &grid, const MagneticField< DeviceType > &magnetic_field, const VelocityGrid &vgrid, Plasma &plasma, DomainDecomposition< DeviceType > &pol_decomp, WeightingAlgorithm weighting_algorithm)

void	redistribute_load (const Simulation< DeviceType > &sml, const Grid< DeviceType > &grid, const MagneticField< DeviceType > &magnetic_field, const VelocityGrid &vgrid, Plasma &plasma, DomainDecomposition< DeviceType > &pol_decomp, const View< int *, CLayout, HostType > &old_partition)

bool	will_rebalance (ReweightOption reweight_option, WeightingAlgorithm weighting_algorithm, int istep, int gstep, double f0_cost)

Private Attributes
WeightingAlgorithm	default_weighting_algorithm

std::vector< LoadRegion >	regions

View< int *, CLayout, HostType >	proposed_partition
	Which processors get which vertices. More...

double	threshold_to_rebalance

bool	verbose

double	constraint1_max

Member Enumeration Documentation

◆ ConstraintOption

enum LoadBalance::ConstraintOption

strong

Enumerator
ParticleCount

◆ ReweightOption

enum LoadBalance::ReweightOption

strong

Enumerator
Always
IfBetter

◆ WeightingAlgorithm

enum LoadBalance::WeightingAlgorithm

strong

Enumerator
SingleRegionBalance
ParticleBalance
Fortran
Default

Constructor & Destructor Documentation

◆ LoadBalance() [1/2]

LoadBalance::LoadBalance ( )

inline

◆ LoadBalance() [2/2]

LoadBalance::LoadBalance	(	NLReader::NamelistReader &	nlr,
		const DomainDecomposition< DeviceType > &	pol_decomp,
		bool	sync_planes = `true`
	)

inline

Here is the call graph for this function:

Member Function Documentation

◆ get_even_division()

double LoadBalance::get_even_division	(	const View< double *, HostType > &	input,
		int	n
	)		const

inlineprivate

Here is the caller graph for this function:

◆ get_largest_predicted_time()

double LoadBalance::get_largest_predicted_time	(	const View< int *, CLayout, HostType > &	partition,
		const View< double *, HostType > &	weight
	)		const

inlineprivate

Here is the caller graph for this function:

◆ greedily_fill_partition()

bool LoadBalance::greedily_fill_partition	(	const View< double *, HostType > &	weight,
		const View< double *, HostType > &	constraint1,
		double	target_weight_per_rank
	)

inlineprivate

Here is the caller graph for this function:

◆ initialize_model()

void LoadBalance::initialize_model ( )

inlineprivate

Here is the caller graph for this function:

◆ model_is_initialized()

bool LoadBalance::model_is_initialized ( )

inlineprivate

Here is the caller graph for this function:

◆ one_weight_balance()

void LoadBalance::one_weight_balance	(	const View< double *, HostType > &	weight,
		const View< double *, CLayout, HostType >	constraint1
	)

inlineprivate

Here is the call graph for this function:

Here is the caller graph for this function:

◆ print_new_partition()

void LoadBalance::print_new_partition ( )

inlineprivate

Here is the caller graph for this function:

◆ propose_new_partition()

void LoadBalance::propose_new_partition	(	const Kokkos::View< double *, Kokkos::LayoutRight, HostType > &	ptl_count,
		WeightingAlgorithm	weighting_algorithm
	)

inlineprivate

Here is the call graph for this function:

Here is the caller graph for this function:

◆ rebalance() [1/2]

void LoadBalance::rebalance	(	const Simulation< DeviceType > &	sml,
		const Grid< DeviceType > &	grid,
		const MagneticField< DeviceType > &	magnetic_field,
		const VelocityGrid &	vgrid,
		Plasma &	plasma,
		DomainDecomposition< DeviceType > &	pol_decomp,
		int	istep,
		int	gstep,
		ReweightOption	reweight_option,
		WeightingAlgorithm	weighting_algorithm = `WeightingAlgorithm::Default`
	)

inline

Here is the call graph for this function:

Here is the caller graph for this function:

◆ rebalance() [2/2]

void LoadBalance::rebalance	(	DomainDecomposition< DeviceType > &	pol_decomp,
		const View< double *, CLayout, HostType > &	constraint,
		const std::vector< double > &	timings,
		double &	load_imbalance,
		View< double *, HostType > &	model_belief
	)

inline

Here is the call graph for this function:

◆ recommend_proposed_partition()

bool LoadBalance::recommend_proposed_partition ( )

inlineprivate

Here is the caller graph for this function:

◆ redistribute_load()

void LoadBalance::redistribute_load	(	const Simulation< DeviceType > &	sml,
		const Grid< DeviceType > &	grid,
		const MagneticField< DeviceType > &	magnetic_field,
		const VelocityGrid &	vgrid,
		Plasma &	plasma,
		DomainDecomposition< DeviceType > &	pol_decomp,
		const View< int *, CLayout, HostType > &	old_partition
	)

inlineprivate

Here is the call graph for this function:

Here is the caller graph for this function:

◆ set_new_partition()

void LoadBalance::set_new_partition	(	const Simulation< DeviceType > &	sml,
		const Grid< DeviceType > &	grid,
		const MagneticField< DeviceType > &	magnetic_field,
		const VelocityGrid &	vgrid,
		Plasma &	plasma,
		DomainDecomposition< DeviceType > &	pol_decomp,
		WeightingAlgorithm	weighting_algorithm
	)

inlineprivate

Here is the call graph for this function:

Here is the caller graph for this function:

◆ update_model() [1/2]

void LoadBalance::update_model ( const View< int *, CLayout, HostType > & current_partition )

inlineprivate

Here is the caller graph for this function:

◆ update_model() [2/2]

void LoadBalance::update_model	(	const View< int *, CLayout, HostType > &	current_partition,
		const std::vector< double > &	manual_times
	)

inlineprivate

◆ will_rebalance()

bool LoadBalance::will_rebalance	(	ReweightOption	reweight_option,
		WeightingAlgorithm	weighting_algorithm,
		int	istep,
		int	gstep,
		double	f0_cost
	)

inlineprivate

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

◆ constraint1_max

double LoadBalance::constraint1_max

private

◆ default_weighting_algorithm

WeightingAlgorithm LoadBalance::default_weighting_algorithm

private

◆ proposed_partition

View<int*,CLayout,HostType> LoadBalance::proposed_partition

private

Which processors get which vertices.

◆ regions

std::vector<LoadRegion> LoadBalance::regions

private

◆ threshold_to_rebalance

double LoadBalance::threshold_to_rebalance

private

◆ verbose

bool LoadBalance::verbose

private

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

/p/test_ssd/builds/t3_84szKM/0/xgc/XGC-Devel/XGC_core/cpp/load_balance.hpp

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Member Enumeration Documentation

◆ ConstraintOption

◆ ReweightOption

◆ WeightingAlgorithm

Constructor & Destructor Documentation

◆ LoadBalance() [1/2]

◆ LoadBalance() [2/2]

Member Function Documentation

◆ get_even_division()

◆ get_largest_predicted_time()

◆ greedily_fill_partition()

◆ initialize_model()

◆ model_is_initialized()

◆ one_weight_balance()

◆ print_new_partition()

◆ propose_new_partition()

◆ rebalance() [1/2]

◆ rebalance() [2/2]

◆ recommend_proposed_partition()

◆ redistribute_load()

◆ set_new_partition()

◆ update_model() [1/2]

◆ update_model() [2/2]

◆ will_rebalance()

Member Data Documentation

◆ constraint1_max

◆ default_weighting_algorithm

◆ proposed_partition

◆ regions

◆ threshold_to_rebalance

◆ verbose