yuxihu commented on a change in pull request #14615: Add MXEnginePushAsync and
MXEnginePushSync C APIs
URL: https://github.com/apache/incubator-mxnet/pull/14615#discussion_r274166971
##########
File path: src/c_api/c_api.cc
##########
@@ -1401,3 +1401,78 @@ int MXNDArrayCreateFromSharedMem(int shared_pid, int
shared_id, const mx_uint *s
*out = new NDArray(shared_pid, shared_id, mxnet::TShape(shape, shape +
ndim), dtype);
API_END();
}
+
+typedef Engine::VarHandle VarHandle;
+typedef Engine::CallbackOnComplete CallbackOnComplete;
+
+int MXEnginePushAsync(EngineAsyncFunc async_func, void* func_param,
+ EngineFuncParamDeleter deleter, ContextHandle ctx_handle,
+ EngineVarHandle const_vars_handle, size_t num_const_vars,
+ EngineVarHandle mutable_vars_handle, size_t
num_mutable_vars,
+ EngineFnPropertyHandle prop_handle, int priority,
+ const char* opr_name, bool wait) {
+ API_BEGIN();
+
+ auto exec_ctx = *static_cast<const Context*>(ctx_handle);
+ auto prop = *static_cast<const FnProperty*>(prop_handle);
+ auto const_vars = static_cast<VarHandle*>(const_vars_handle);
+ auto mutable_vars = static_cast<VarHandle*>(mutable_vars_handle);
+
+ Engine::AsyncFn exec_fn;
+ if (deleter == nullptr) {
+ exec_fn = [async_func, func_param](RunContext rctx,
+ CallbackOnComplete on_complete) {
+ async_func(&rctx, &on_complete, func_param);
+ };
+ } else {
+ // Wrap func_param in a shared_ptr with del as deleter such that delelter
Review comment:
Done
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
