#include <api_impl.h>

class ClientInstance

ClientInstance(std::unique_ptr< Platform > platform)

virtual ~ClientInstance()

PJRT_Error * Initialize()

inline Platform & platform()

inline Logger & logger()

inline iree_allocator_t host_allocator()

inline const std::vector< DeviceInstance * > & devices()

inline const std::vector< DeviceInstance * > & addressable_devices()

inline const std::vector< MemoryInstance * > & memories()

inline const std::string & cached_platform_name()

inline const std::string & cached_platform_version()

inline iree_vm_instance_t * vm_instance()

PJRT_Error * Compile(const PJRT_Program * program, xla::CompileOptionsProto options, LoadedExecutableInstance ** executable)

virtual iree_status_t CreateDriver(iree_hal_driver_t ** out_driver)

virtual iree_status_t PopulateVMModules(std::vector< iree::vm::ref< iree_vm_module_t > > & modules, iree_hal_device_t * hal_device, iree::vm::ref< iree_vm_module_t > & main_module)

virtual bool SetDefaultCompilerFlags(CompilerJob * compiler_job)

std::tuple< uint64_t, uint64_t > AdvanceTimeline()

static void BindApi(PJRT_Api * api)

static inline ClientInstance * Unwrap(PJRT_Client * client)

iree_allocator_t host_allocator_

iree_hal_driver_registry_t * driver_registry_ = nullptr

std::string cached_platform_name_

std::string cached_platform_version_

std::unique_ptr< Platform > platform_

iree_hal_driver_t * driver_ = nullptr

iree_hal_device_info_t * device_infos_ = nullptr

iree_host_size_t device_info_count_ = 0

std::vector< DeviceInstance * > devices_

std::vector< DeviceInstance * > addressable_devices_

std::vector< MemoryInstance * > memories_

iree::vm::ref< iree_vm_instance_t > vm_instance_

uint64_t execution_timeline_ = 0ull

iree_status_t InitializeCompiler()

iree_status_t InitializeVM()

iree_status_t PopulateDevices()