dereferencing dev_type

apache · Aug 29, 2019 · f44e64a · f44e64a
1 parent 7045b09
commit f44e64a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/include/mxnet/c_api.h b/include/mxnet/c_api.h
@@ -2016,7 +2016,7 @@ MXNET_DLL int MXGenAtomicSymbolFromSymbol(SymbolHandle sym_handle, SymbolHandle
  */
 MXNET_DLL int MXOptimizeForBackend(SymbolHandle sym_handle,
                                    const char* backend_name,
-                                   const int* dev_type,
+                                   const int dev_type,
                                    SymbolHandle* ret_sym_handle,
                                    const mx_uint len,
                                    NDArrayHandle* in_args_handle,

diff --git a/src/c_api/c_api_symbolic.cc b/src/c_api/c_api_symbolic.cc
@@ -1218,7 +1218,7 @@ int MXShallowCopySymbol(SymbolHandle src, SymbolHandle* out) {
 
 int MXOptimizeForBackend(SymbolHandle sym_handle,
                          const char* backend_name,
-                         const int* dev_type,
+                         const int dev_type,
                          SymbolHandle* ret_sym_handle,
                          const mx_uint len,
                          NDArrayHandle* in_args_handle,
@@ -1232,7 +1232,7 @@ int MXOptimizeForBackend(SymbolHandle sym_handle,
   nnvm::Graph g = Symbol2Graph(*s);
   if (len) {
     NDArray **in_args_ptr = reinterpret_cast<NDArray**>(in_args_handle);
-    Context default_ctx = Context::Create(static_cast<Context::DeviceType>(*dev_type), 0);
+    Context default_ctx = Context::Create(static_cast<Context::DeviceType>(dev_type), 0);
     mxnet::ShapeVector arg_shapes(len);
     nnvm::DTypeVector arg_dtypes(len);
     StorageTypeVector arg_stypes(len);