30 using namespace mlir::runtime;
37 namespace runtime {  namespace
126   RefCounted(AsyncRuntime *runtime, int64_t refCount = 1)  in RefCounted()  argument
127       : runtime(runtime), refCount(refCount) {  in RefCounted()
128     runtime->addNumRefCountedObjects();  in RefCounted()
133     runtime->dropNumRefCountedObjects();  in ~RefCounted()
152   AsyncRuntime *runtime;  member in mlir::runtime::__anonc7b155b40111::RefCounted
160   static auto runtime = std::make_unique<AsyncRuntime>();  in getDefaultAsyncRuntimeInstance()  local
161   return runtime;  in getDefaultAsyncRuntimeInstance()
179   AsyncToken(AsyncRuntime *runtime)  in AsyncToken()
180       : RefCounted(runtime, /*refCount=*/2), state(State::kUnavailable) {}  in AsyncToken()
195   AsyncValue(AsyncRuntime *runtime, int64_t size)  in AsyncValue()
196       : RefCounted(runtime, /*refCount=*/2), state(State::kUnavailable),  in AsyncValue()
214   AsyncGroup(AsyncRuntime *runtime, int64_t size)  in AsyncGroup()
215       : RefCounted(runtime), pendingTokens(size), numErrors(0), rank(0) {}  in AsyncGroup()
398   auto *runtime = getDefaultAsyncRuntime();  in mlirAsyncRuntimeExecute()  local
399   runtime->getThreadPool().async([handle, resume]() { (*resume)(handle); });  in mlirAsyncRuntimeExecute()
481                &mlir::runtime::mlirAsyncRuntimeAddRef);  in __mlir_runner_init()
483                &mlir::runtime::mlirAsyncRuntimeDropRef);  in __mlir_runner_init()
485                &mlir::runtime::mlirAsyncRuntimeExecute);  in __mlir_runner_init()
487                &mlir::runtime::mlirAsyncRuntimeGetValueStorage);  in __mlir_runner_init()
489                &mlir::runtime::mlirAsyncRuntimeCreateToken);  in __mlir_runner_init()
491                &mlir::runtime::mlirAsyncRuntimeCreateValue);  in __mlir_runner_init()
493                &mlir::runtime::mlirAsyncRuntimeEmplaceToken);  in __mlir_runner_init()
495                &mlir::runtime::mlirAsyncRuntimeEmplaceValue);  in __mlir_runner_init()
497                &mlir::runtime::mlirAsyncRuntimeSetTokenError);  in __mlir_runner_init()
499                &mlir::runtime::mlirAsyncRuntimeSetValueError);  in __mlir_runner_init()
501                &mlir::runtime::mlirAsyncRuntimeIsTokenError);  in __mlir_runner_init()
503                &mlir::runtime::mlirAsyncRuntimeIsValueError);  in __mlir_runner_init()
505                &mlir::runtime::mlirAsyncRuntimeIsGroupError);  in __mlir_runner_init()
507                &mlir::runtime::mlirAsyncRuntimeAwaitToken);  in __mlir_runner_init()
509                &mlir::runtime::mlirAsyncRuntimeAwaitValue);  in __mlir_runner_init()
511                &mlir::runtime::mlirAsyncRuntimeAwaitTokenAndExecute);  in __mlir_runner_init()
513                &mlir::runtime::mlirAsyncRuntimeAwaitValueAndExecute);  in __mlir_runner_init()
515                &mlir::runtime::mlirAsyncRuntimeCreateGroup);  in __mlir_runner_init()
517                &mlir::runtime::mlirAsyncRuntimeAddTokenToGroup);  in __mlir_runner_init()
519                &mlir::runtime::mlirAsyncRuntimeAwaitAllInGroup);  in __mlir_runner_init()
521                &mlir::runtime::mlirAsyncRuntimeAwaitAllInGroupAndExecute);  in __mlir_runner_init()
523                &mlir::runtime::mlirAsyncRuntimGetNumWorkerThreads);  in __mlir_runner_init()
525                &mlir::runtime::mlirAsyncRuntimePrintCurrentThreadId);  in __mlir_runner_init()