Merge pull request #2 from mavenlin/master

fix python call bug
apache · Aug 20, 2015 · c956ae7 · c956ae7
2 parents 910738d + 72eb074
commit c956ae7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/python/mxnet/function.py b/python/mxnet/function.py
@@ -124,7 +124,7 @@ def __init__(self):
                                         ctypes.byref(plist)))
         hmap = {}
         for i in range(size.value):
-            hdl = plist[i]
+            hdl = ctypes.c_void_p(plist[i])
             name = ctypes.c_char_p()
             check_call(_LIB.MXFuncGetName(hdl, ctypes.byref(name)))
             hmap[name.value] = _Function(hdl, name.value)

diff --git a/python/mxnet/symbol_creator.py b/python/mxnet/symbol_creator.py
@@ -83,9 +83,10 @@ def __init__(self):
                                                          ctypes.byref(plist)))
         hmap = {}
         for i in range(size.value):
+            hdl = ctypes.c_void_p(plist[i])
             name = ctypes.c_char_p()
-            check_call(_LIB.MXSymbolGetAtomicSymbolName(plist[i], ctypes.byref(name)))
-            hmap[name.value] = _SymbolCreator(name, plist[i])
+            check_call(_LIB.MXSymbolGetAtomicSymbolName(hdl, ctypes.byref(name)))
+            hmap[name.value] = _SymbolCreator(name, hdl)
         self.__dict__.update(hmap)
 
     def Variable(self, name):