modify the type of arguments in PushSync and PushAsync

apache · Aug 7, 2019 · ee98c12 · ee98c12
1 parent 14d4eb9
commit ee98c12
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/include/mxnet/c_api.h b/include/mxnet/c_api.h
@@ -2875,8 +2875,8 @@ MXNET_DLL int MXNDArrayCreateFromSharedMemEx(int shared_pid, int shared_id, cons
   */
 MXNET_DLL int MXEnginePushAsync(EngineAsyncFunc async_func, void* func_param,
                                 EngineFuncParamDeleter deleter, ContextHandle ctx_handle,
-                                EngineVarHandle const_vars_handle, int num_const_vars,
-                                EngineVarHandle mutable_vars_handle, int num_mutable_vars,
+                                EngineVarHandle* const_vars_handle, int num_const_vars,
+                                EngineVarHandle* mutable_vars_handle, int num_mutable_vars,
                                 EngineFnPropertyHandle prop_handle DEFAULT(NULL),
                                 int priority DEFAULT(0), const char* opr_name DEFAULT(NULL),
                                 bool wait DEFAULT(false));
@@ -2898,8 +2898,8 @@ MXNET_DLL int MXEnginePushAsync(EngineAsyncFunc async_func, void* func_param,
   */
 MXNET_DLL int MXEnginePushSync(EngineSyncFunc sync_func, void* func_param,
                                EngineFuncParamDeleter deleter, ContextHandle ctx_handle,
-                               EngineVarHandle const_vars_handle, int num_const_vars,
-                               EngineVarHandle mutable_vars_handle, int num_mutable_vars,
+                               EngineVarHandle* const_vars_handle, int num_const_vars,
+                               EngineVarHandle* mutable_vars_handle, int num_mutable_vars,
                                EngineFnPropertyHandle prop_handle DEFAULT(NULL),
                                int priority DEFAULT(0), const char* opr_name DEFAULT(NULL));
 

diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc
@@ -1465,8 +1465,8 @@ void AssertValidNumberVars(int num_const_vars, int num_mutable_vars) {
 
 int MXEnginePushAsync(EngineAsyncFunc async_func, void* func_param,
                       EngineFuncParamDeleter deleter, ContextHandle ctx_handle,
-                      EngineVarHandle const_vars_handle, int num_const_vars,
-                      EngineVarHandle mutable_vars_handle, int num_mutable_vars,
+                      EngineVarHandle* const_vars_handle, int num_const_vars,
+                      EngineVarHandle* mutable_vars_handle, int num_mutable_vars,
                       EngineFnPropertyHandle prop_handle, int priority,
                       const char* opr_name, bool wait) {
   API_BEGIN();
@@ -1506,8 +1506,8 @@ int MXEnginePushAsync(EngineAsyncFunc async_func, void* func_param,
 
 int MXEnginePushSync(EngineSyncFunc sync_func, void* func_param,
                      EngineFuncParamDeleter deleter, ContextHandle ctx_handle,
-                     EngineVarHandle const_vars_handle, int num_const_vars,
-                     EngineVarHandle mutable_vars_handle, int num_mutable_vars,
+                     EngineVarHandle* const_vars_handle, int num_const_vars,
+                     EngineVarHandle* mutable_vars_handle, int num_mutable_vars,
                      EngineFnPropertyHandle prop_handle, int priority,
                      const char* opr_name) {
   API_BEGIN();