pjrt

EventInstance

EventInstance

#include <api_impl.h>
class EventInstance

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:317

Public Methods


EventInstance

EventInstance(iree::vm::ref< iree_hal_fence_t > fence)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:319


~EventInstance

~EventInstance()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:320


operator PJRT_Event *

inline

inline operator PJRT_Event *()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:321


OnReady

iree_status_t OnReady(PJRT_Event_OnReadyCallback callback, void * user_arg)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:327


error

ErrorInstance * error()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:328


is_ready

bool is_ready()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:329

Public Static Methods

ReturnNameDescription
voidBindApi static
EventInstance *Unwrap static inline

BindApi

static

static void BindApi(PJRT_Api * api)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:322


Unwrap

static inline

static inline EventInstance * Unwrap(PJRT_Event * exe)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:323

Private Attributes

ReturnNameDescription
std::mutexlock_
iree_status_tstatus_
boolis_ready_
std::vector< std::pair< PJRT_Event_OnReadyCallback, void * > >pending_callbacks_
std::unique_ptr< std::thread >signal_thread_

lock_

std::mutex lock_

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:334


status_

iree_status_t status_ = iree_ok_status()

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:335


is_ready_

bool is_ready_

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:336


pending_callbacks_

std::vector< std::pair< PJRT_Event_OnReadyCallback, void * > > pending_callbacks_

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:337


signal_thread_

std::unique_ptr< std::thread > signal_thread_

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:338

Private Methods

ReturnNameDescription
voidSignalReady

SignalReady

void SignalReady(iree_status_t status)

Defined in integrations/pjrt/src/iree_pjrt/common/api_impl.h:332