load_balance is a strategy_type which uses the load balance algorithm.
More...
#include <ginkgo/core/matrix/csr.hpp>
template<
typename ValueType = default_precision,
typename IndexType = int32>
class gko::matrix::Csr< ValueType, IndexType >::load_balance
load_balance is a strategy_type which uses the load balance algorithm.
◆ load_balance() [1/5]
Creates a load_balance strategy.
- Warning
- this is deprecated! Please rely on the new automatic strategy instantiation or use one of the other constructors.
◆ load_balance() [2/5]
Creates a load_balance strategy with CUDA executor.
- Parameters
-
◆ load_balance() [3/5]
Creates a load_balance strategy with HIP executor.
- Parameters
-
◆ load_balance() [4/5]
Creates a load_balance strategy with DPCPP executor.
- Parameters
-
- Note
- TODO: porting - we hardcode the subgroup size is 32
◆ load_balance() [5/5]
gko::matrix::Csr< ValueType, IndexType >::load_balance::load_balance |
( |
int64_t | nwarps, |
|
|
int | warp_size = 32, |
|
|
bool | cuda_strategy = true, |
|
|
std::string | strategy_name = "none" ) |
|
inline |
Creates a load_balance strategy with specified parameters.
- Parameters
-
nwarps | the number of warps in the executor |
warp_size | the warp size of the executor |
cuda_strategy | whether the cuda_strategy needs to be used. |
- Note
- The warp_size must be the size of full warp. When using this constructor, set_strategy needs to be called with correct parameters which is replaced during the conversion.
◆ clac_size()
◆ copy()
◆ process()
The documentation for this class was generated from the following file: