This is the Executor subclass which represents the HIP enhanced device.
More...
#include <ginkgo/core/base/executor.hpp>
This is the Executor subclass which represents the HIP enhanced device.
◆ create()
static std::shared_ptr<HipExecutor> gko::HipExecutor::create |
( |
int |
device_id, |
|
|
std::shared_ptr< Executor > |
master, |
|
|
bool |
device_reset, |
|
|
allocation_mode |
alloc_mode = default_hip_alloc_mode , |
|
|
CUstream_st * |
stream = nullptr |
|
) |
| |
|
static |
Creates a new HipExecutor.
- Parameters
-
device_id | the HIP device id of this device |
master | an executor on the host that is used to invoke the device kernels |
device_reset | whether to reset the device after the object exits the scope. |
alloc_mode | the allocation mode that the executor should operate on. See @allocation_mode for more details |
◆ get_blas_handle()
hipblasContext* gko::HipExecutor::get_blas_handle |
( |
| ) |
const |
|
inline |
Get the hipblas handle for this executor.
- Returns
- the hipblas handle (hipblasContext*) for this executor
Referenced by get_hipblas_handle().
◆ get_closest_numa()
int gko::HipExecutor::get_closest_numa |
( |
| ) |
const |
|
inline |
Get the closest NUMA node.
- Returns
- the closest NUMA node closest to this device
◆ get_closest_pus()
std::vector<int> gko::HipExecutor::get_closest_pus |
( |
| ) |
const |
|
inline |
Get the closest PUs.
- Returns
- the array of PUs closest to this device
◆ get_description()
std::string gko::HipExecutor::get_description |
( |
| ) |
const |
|
overridevirtual |
- Returns
- a textual representation of the executor and its device.
Implements gko::Executor.
◆ get_hipblas_handle()
hipblasContext* gko::HipExecutor::get_hipblas_handle |
( |
| ) |
const |
|
inline |
Get the hipblas handle for this executor.
- Returns
- the hipblas handle (hipblasContext*) for this executor
References get_blas_handle().
◆ get_hipsparse_handle()
hipsparseContext* gko::HipExecutor::get_hipsparse_handle |
( |
| ) |
const |
|
inline |
Get the hipsparse handle for this executor.
- Returns
- the hipsparse handle (hipsparseContext*) for this executor
References get_sparselib_handle().
◆ get_master() [1/2]
std::shared_ptr<const Executor> gko::HipExecutor::get_master |
( |
| ) |
const |
|
overridevirtualnoexcept |
◆ get_master() [2/2]
std::shared_ptr<Executor> gko::HipExecutor::get_master |
( |
| ) |
|
|
overridevirtualnoexcept |
◆ get_sparselib_handle()
hipsparseContext* gko::HipExecutor::get_sparselib_handle |
( |
| ) |
const |
|
inline |
Get the hipsparse handle for this executor.
- Returns
- the hipsparse handle (hipsparseContext*) for this executor
Referenced by get_hipsparse_handle().
The documentation for this class was generated from the following file: