MLIR  20.0.0git
Classes | Typedefs | Functions
mlir::runtime Namespace Reference

Classes

struct  AsyncToken
 
struct  AsyncValue
 
struct  AsyncGroup
 

Typedefs

using AsyncToken = struct AsyncToken
 
using AsyncGroup = struct AsyncGroup
 
using AsyncValue = struct AsyncValue
 
using ValueStorage = std::byte *
 
using CoroHandle = void *
 
using CoroResume = void(*)(void *)
 
using RefCountedObjPtr = void *
 

Functions

MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeAddRef (RefCountedObjPtr, int64_t)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeDropRef (RefCountedObjPtr, int64_t)
 
MLIR_ASYNC_RUNTIME_EXPORT AsyncTokenmlirAsyncRuntimeCreateToken ()
 
MLIR_ASYNC_RUNTIME_EXPORT AsyncValuemlirAsyncRuntimeCreateValue (int64_t)
 
MLIR_ASYNC_RUNTIME_EXPORT AsyncGroupmlirAsyncRuntimeCreateGroup (int64_t size)
 
MLIR_ASYNC_RUNTIME_EXPORT int64_t mlirAsyncRuntimeAddTokenToGroup (AsyncToken *, AsyncGroup *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeEmplaceToken (AsyncToken *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeEmplaceValue (AsyncValue *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeSetTokenError (AsyncToken *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeSetValueError (AsyncValue *)
 
MLIR_ASYNC_RUNTIME_EXPORT bool mlirAsyncRuntimeIsTokenError (AsyncToken *)
 
MLIR_ASYNC_RUNTIME_EXPORT bool mlirAsyncRuntimeIsValueError (AsyncValue *)
 
MLIR_ASYNC_RUNTIME_EXPORT bool mlirAsyncRuntimeIsGroupError (AsyncGroup *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeAwaitToken (AsyncToken *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeAwaitValue (AsyncValue *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeAwaitAllInGroup (AsyncGroup *)
 
MLIR_ASYNC_RUNTIME_EXPORT ValueStorage mlirAsyncRuntimeGetValueStorage (AsyncValue *)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeExecute (CoroHandle, CoroResume)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeAwaitTokenAndExecute (AsyncToken *, CoroHandle, CoroResume)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeAwaitValueAndExecute (AsyncValue *, CoroHandle, CoroResume)
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimeAwaitAllInGroupAndExecute (AsyncGroup *, CoroHandle, CoroResume)
 
MLIR_ASYNC_RUNTIME_EXPORT int64_t mlirAsyncRuntimGetNumWorkerThreads ()
 
MLIR_ASYNC_RUNTIME_EXPORT void mlirAsyncRuntimePrintCurrentThreadId ()
 
static std::unique_ptr< AsyncRuntime > & getDefaultAsyncRuntimeInstance ()
 
static void resetDefaultAsyncRuntime ()
 
static AsyncRuntime * getDefaultAsyncRuntime ()
 
static void setTokenState (AsyncToken *token, State state)
 
static void setValueState (AsyncValue *value, State state)
 
MLIR_ASYNC_RUNTIME_EXPORT void __mlir_execution_engine_init (llvm::StringMap< void * > &exportSymbols)
 
MLIR_ASYNC_RUNTIME_EXPORT void __mlir_execution_engine_destroy ()
 

Typedef Documentation

◆ AsyncGroup

using mlir::runtime::AsyncGroup = typedef struct AsyncGroup

Definition at line 46 of file AsyncRuntime.h.

◆ AsyncToken

using mlir::runtime::AsyncToken = typedef struct AsyncToken

Definition at line 43 of file AsyncRuntime.h.

◆ AsyncValue

using mlir::runtime::AsyncValue = typedef struct AsyncValue

Definition at line 49 of file AsyncRuntime.h.

◆ CoroHandle

using mlir::runtime::CoroHandle = typedef void *

Definition at line 57 of file AsyncRuntime.h.

◆ CoroResume

using mlir::runtime::CoroResume = typedef void (*)(void *)

Definition at line 58 of file AsyncRuntime.h.

◆ RefCountedObjPtr

using mlir::runtime::RefCountedObjPtr = typedef void *

Definition at line 62 of file AsyncRuntime.h.

◆ ValueStorage

using mlir::runtime::ValueStorage = typedef std::byte *

Definition at line 52 of file AsyncRuntime.h.

Function Documentation

◆ __mlir_execution_engine_destroy()

MLIR_ASYNC_RUNTIME_EXPORT void mlir::runtime::__mlir_execution_engine_destroy ( )

Definition at line 521 of file AsyncRuntime.cpp.

References resetDefaultAsyncRuntime().

◆ __mlir_execution_engine_init()

void mlir::runtime::__mlir_execution_engine_init ( llvm::StringMap< void * > &  exportSymbols)

◆ getDefaultAsyncRuntime()

static AsyncRuntime* mlir::runtime::getDefaultAsyncRuntime ( )
static

◆ getDefaultAsyncRuntimeInstance()

static std::unique_ptr<AsyncRuntime>& mlir::runtime::getDefaultAsyncRuntimeInstance ( )
static

Definition at line 157 of file AsyncRuntime.cpp.

Referenced by getDefaultAsyncRuntime(), and resetDefaultAsyncRuntime().

◆ mlirAsyncRuntimeAddRef()

void mlir::runtime::mlirAsyncRuntimeAddRef ( RefCountedObjPtr  ptr,
int64_t  count 
)

Definition at line 226 of file AsyncRuntime.cpp.

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeAddTokenToGroup()

int64_t mlir::runtime::mlirAsyncRuntimeAddTokenToGroup ( AsyncToken token,
AsyncGroup group 
)

◆ mlirAsyncRuntimeAwaitAllInGroup()

void mlir::runtime::mlirAsyncRuntimeAwaitAllInGroup ( AsyncGroup group)

◆ mlirAsyncRuntimeAwaitAllInGroupAndExecute()

void mlir::runtime::mlirAsyncRuntimeAwaitAllInGroupAndExecute ( AsyncGroup group,
CoroHandle  handle,
CoroResume  resume 
)

◆ mlirAsyncRuntimeAwaitToken()

void mlir::runtime::mlirAsyncRuntimeAwaitToken ( AsyncToken token)

◆ mlirAsyncRuntimeAwaitTokenAndExecute()

void mlir::runtime::mlirAsyncRuntimeAwaitTokenAndExecute ( AsyncToken token,
CoroHandle  handle,
CoroResume  resume 
)

◆ mlirAsyncRuntimeAwaitValue()

void mlir::runtime::mlirAsyncRuntimeAwaitValue ( AsyncValue value)

◆ mlirAsyncRuntimeAwaitValueAndExecute()

void mlir::runtime::mlirAsyncRuntimeAwaitValueAndExecute ( AsyncValue value,
CoroHandle  handle,
CoroResume  resume 
)

◆ mlirAsyncRuntimeCreateGroup()

AsyncGroup * mlir::runtime::mlirAsyncRuntimeCreateGroup ( int64_t  size)

Definition at line 250 of file AsyncRuntime.cpp.

References getDefaultAsyncRuntime().

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeCreateToken()

AsyncToken * mlir::runtime::mlirAsyncRuntimeCreateToken ( )

Definition at line 238 of file AsyncRuntime.cpp.

References getDefaultAsyncRuntime().

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeCreateValue()

AsyncValue * mlir::runtime::mlirAsyncRuntimeCreateValue ( int64_t  size)

Definition at line 244 of file AsyncRuntime.cpp.

References getDefaultAsyncRuntime().

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeDropRef()

void mlir::runtime::mlirAsyncRuntimeDropRef ( RefCountedObjPtr  ptr,
int64_t  count 
)

Definition at line 232 of file AsyncRuntime.cpp.

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeEmplaceToken()

void mlir::runtime::mlirAsyncRuntimeEmplaceToken ( AsyncToken token)

Definition at line 341 of file AsyncRuntime.cpp.

References setTokenState().

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeEmplaceValue()

void mlir::runtime::mlirAsyncRuntimeEmplaceValue ( AsyncValue value)

Definition at line 345 of file AsyncRuntime.cpp.

References setValueState().

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeExecute()

void mlir::runtime::mlirAsyncRuntimeExecute ( CoroHandle  handle,
CoroResume  resume 
)

Definition at line 395 of file AsyncRuntime.cpp.

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeGetValueStorage()

ValueStorage mlir::runtime::mlirAsyncRuntimeGetValueStorage ( AsyncValue value)

◆ mlirAsyncRuntimeIsGroupError()

bool mlir::runtime::mlirAsyncRuntimeIsGroupError ( AsyncGroup group)

Definition at line 365 of file AsyncRuntime.cpp.

References mlir::runtime::AsyncGroup::numErrors.

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeIsTokenError()

bool mlir::runtime::mlirAsyncRuntimeIsTokenError ( AsyncToken token)

Definition at line 357 of file AsyncRuntime.cpp.

References mlir::runtime::AsyncToken::state.

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeIsValueError()

bool mlir::runtime::mlirAsyncRuntimeIsValueError ( AsyncValue value)

Definition at line 361 of file AsyncRuntime.cpp.

References mlir::runtime::AsyncValue::state.

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimePrintCurrentThreadId()

void mlir::runtime::mlirAsyncRuntimePrintCurrentThreadId ( )

Definition at line 447 of file AsyncRuntime.cpp.

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeSetTokenError()

void mlir::runtime::mlirAsyncRuntimeSetTokenError ( AsyncToken token)

Definition at line 349 of file AsyncRuntime.cpp.

References setTokenState().

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimeSetValueError()

void mlir::runtime::mlirAsyncRuntimeSetValueError ( AsyncValue value)

Definition at line 353 of file AsyncRuntime.cpp.

References setValueState().

Referenced by __mlir_execution_engine_init().

◆ mlirAsyncRuntimGetNumWorkerThreads()

int64_t mlir::runtime::mlirAsyncRuntimGetNumWorkerThreads ( )

Definition at line 439 of file AsyncRuntime.cpp.

References getDefaultAsyncRuntime().

Referenced by __mlir_execution_engine_init().

◆ resetDefaultAsyncRuntime()

static void mlir::runtime::resetDefaultAsyncRuntime ( )
static

Definition at line 162 of file AsyncRuntime.cpp.

References getDefaultAsyncRuntimeInstance().

Referenced by __mlir_execution_engine_destroy().

◆ setTokenState()

static void mlir::runtime::setTokenState ( AsyncToken token,
State  state 
)
static

◆ setValueState()

static void mlir::runtime::setValueState ( AsyncValue value,
State  state 
)
static