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_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_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
◆ 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
◆ 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 |
The documentation for this class was generated from the following file: