Kan Wu created MXNET-1415:
-----------------------------
Summary: Add MXEnginePushAsyncND and MXEnginePushSyncND C APIs
Key: MXNET-1415
URL: https://issues.apache.org/jira/browse/MXNET-1415
Project: Apache MXNet
Issue Type: Improvement
Components: Apache MXNet C/C++ API
Reporter: Kan Wu
Assignee: Amol Lele
Hi, there.
In the PR #14615, @yuxihu added Add MXEnginePushAsync and MXEnginePushSync C
APIs.
MXEnginePushAsync and MXEnginePushSync needs VarHandles of NDArrays.
However, MXNet doesn't expose the API to get the VarHandle of a NDArray.
After the discussion with @szha, We considered that it's more elegant to add
MXEnginePushAsyncND and MXEnginePushSyncND C APIs, whose arguments are
NDArrayHandles, rather than VarHandles.
https://github.com/apache/incubator-mxnet/pull/15177
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
