load_balance is a strategy_type which uses the load balance algorithm.
More...
#include <ginkgo/core/matrix/csr.hpp>
template<typename ValueType, typename IndexType>
class gko::matrix::Csr< ValueType, IndexType >::load_balance
load_balance is a strategy_type which uses the load balance algorithm.
◆ load_balance() [1/5]
template<typename ValueType , typename IndexType >
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]
template<typename ValueType , typename IndexType >
Creates a load_balance strategy with CUDA executor.
- Parameters
-
◆ load_balance() [3/5]
template<typename ValueType , typename IndexType >
Creates a load_balance strategy with HIP executor.
- Parameters
-
◆ load_balance() [4/5]
template<typename ValueType , typename IndexType >
Creates a load_balance strategy with DPCPP executor.
- Parameters
-
- Note
- TODO: porting - we hardcode the subgroup size is 32
◆ load_balance() [5/5]
template<typename ValueType , typename IndexType >
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()
template<typename ValueType , typename IndexType >
int64_t gko::matrix::Csr< ValueType, IndexType >::load_balance::clac_size |
( |
const int64_t | nnz | ) |
|
|
inlineoverridevirtual |
◆ copy()
template<typename ValueType , typename IndexType >
◆ process()
template<typename ValueType , typename IndexType >
void gko::matrix::Csr< ValueType, IndexType >::load_balance::process |
( |
const array< index_type > & | mtx_row_ptrs, |
|
|
array< index_type > * | mtx_srow ) |
|
inlineoverridevirtual |
The documentation for this class was generated from the following file: