From ee98c12f637ea3a370a99f0fb333562abe31f75f Mon Sep 17 00:00:00 2001 From: wkcn Date: Thu, 8 Aug 2019 07:14:39 +0800 Subject: [PATCH] modify the type of arguments in PushSync and PushAsync --- include/mxnet/c_api.h | 8 ++++---- src/c_api/c_api.cc | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/include/mxnet/c_api.h b/include/mxnet/c_api.h index 1298b12b66ac..0fcabdd01fe9 100644 --- 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 index a7a80a5ab40c..ffc9882b06c3 100644 --- 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();