cpu::CPUClientInstance
CPUClientInstance
#include <client.h>class CPUClientInstanceDefined in integrations/pjrt/src/iree_pjrt/cpu/client.h:16
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 |
|---|---|---|
CPUClientInstance | ||
~CPUClientInstance | ||
iree_status_t | CreateDriver virtual override | |
bool | SetDefaultCompilerFlags virtual override |
CPUClientInstance
CPUClientInstance(std::unique_ptr< Platform > platform)Defined in integrations/pjrt/src/iree_pjrt/cpu/client.h:18
~CPUClientInstance
~CPUClientInstance()Defined in integrations/pjrt/src/iree_pjrt/cpu/client.h:19
CreateDriver
virtual override
virtual iree_status_t CreateDriver(iree_hal_driver_t ** out_driver) overrideDefined in integrations/pjrt/src/iree_pjrt/cpu/client.h:20
Reimplements
SetDefaultCompilerFlags
virtual override
virtual bool SetDefaultCompilerFlags(CompilerJob * compiler_job) overrideDefined in integrations/pjrt/src/iree_pjrt/cpu/client.h:21
Reimplements
Private Attributes
| Return | Name | Description |
|---|---|---|
bool | single_threaded_debug_ | |
iree_task_topology_t | task_topology_options_ | |
iree_hal_executable_plugin_manager_t * | plugin_manager_ | |
iree_hal_executable_loader_t * | loaders_ | |
iree_host_size_t | loader_count_ | |
iree_task_executor_t * | executor_ | |
iree_hal_allocator_t * | device_allocator_ |
single_threaded_debug_
bool single_threaded_debug_ = falseDefined in integrations/pjrt/src/iree_pjrt/cpu/client.h:27
task_topology_options_
iree_task_topology_t task_topology_options_Defined in integrations/pjrt/src/iree_pjrt/cpu/client.h:28
plugin_manager_
iree_hal_executable_plugin_manager_t * plugin_manager_ = nullptrDefined in integrations/pjrt/src/iree_pjrt/cpu/client.h:31
loaders_
iree_hal_executable_loader_t * loaders_[8]Defined in integrations/pjrt/src/iree_pjrt/cpu/client.h:32
loader_count_
iree_host_size_t loader_count_ = 0Defined in integrations/pjrt/src/iree_pjrt/cpu/client.h:33
executor_
iree_task_executor_t * executor_ = nullptrDefined in integrations/pjrt/src/iree_pjrt/cpu/client.h:34
device_allocator_
iree_hal_allocator_t * device_allocator_ = nullptrDefined in integrations/pjrt/src/iree_pjrt/cpu/client.h:35
Private Methods
| Return | Name | Description |
|---|---|---|
iree_status_t | InitializeDeps |
InitializeDeps
iree_status_t InitializeDeps()