rocm
rocm::ROCMClientInstance
ROCMClientInstance
#include <client.h>class ROCMClientInstanceDefined in integrations/pjrt/src/iree_pjrt/rocm/client.h:15
Inherits:
ClientInstance
List of all members
Inherited from ClientInstance
| Kind | Name | Description |
|---|---|---|
function | ClientInstance | |
function | ~ClientInstance virtual | |
function | Initialize | |
function | platform inline | |
function | logger inline | |
function | host_allocator inline | |
function | devices inline | |
function | addressable_devices inline | |
function | memories inline | |
function | cached_platform_name inline | |
function | cached_platform_version inline | |
function | vm_instance inline | |
function | Compile | |
function | CreateDriver virtual | |
function | PopulateVMModules virtual | |
function | SetDefaultCompilerFlags virtual | |
function | AdvanceTimeline | |
function | BindApi static | |
function | Unwrap static inline | |
variable | host_allocator_ | |
variable | driver_registry_ | |
variable | cached_platform_name_ | |
variable | cached_platform_version_ | |
variable | platform_ | |
variable | driver_ | |
variable | device_infos_ | |
variable | device_info_count_ | |
variable | devices_ | |
variable | addressable_devices_ | |
variable | memories_ | |
variable | vm_instance_ | |
variable | execution_timeline_ | |
function | InitializeCompiler | |
function | InitializeVM | |
function | PopulateDevices |
Public Methods
| Return | Name | Description |
|---|---|---|
ROCMClientInstance | ||
~ROCMClientInstance | ||
iree_status_t | CreateDriver virtual override | |
bool | SetDefaultCompilerFlags virtual override |
ROCMClientInstance
ROCMClientInstance(std::unique_ptr< Platform > platform)Defined in integrations/pjrt/src/iree_pjrt/rocm/client.h:17
~ROCMClientInstance
~ROCMClientInstance()Defined in integrations/pjrt/src/iree_pjrt/rocm/client.h:18
CreateDriver
virtual override
virtual iree_status_t CreateDriver(iree_hal_driver_t ** out_driver) overrideDefined in integrations/pjrt/src/iree_pjrt/rocm/client.h:19
Reimplements
SetDefaultCompilerFlags
virtual override
virtual bool SetDefaultCompilerFlags(CompilerJob * compiler_job) overrideDefined in integrations/pjrt/src/iree_pjrt/rocm/client.h:20
