Skip to content

T
tic
Commit 7250005d by Haichen Shen Committed by Tianqi Chen
Options

fix ctypes bug for mac (#9) 

* fix c_api bug for mac

* update travis
parent 39314014
Showing with 9 additions and 7 deletions
.travis.yml
...@@ -4,7 +4,7 @@ language: cpp ...@@ -4,7 +4,7 @@ language: cpp
os: os:
- linux - linux
# - osx - osx
env: env:
# code analysis # code analysis
......
python/tvm/_ctypes/_api.py
...@@ -25,8 +25,13 @@ kDouble = 2 ...@@ -25,8 +25,13 @@ kDouble = 2
kStr = 3 kStr = 3
kNodeHandle = 4 kNodeHandle = 4
NODE_TYPE = {
}
def _type_key(handle): def _return_node(x):
handle = x.v_handle
if not isinstance(handle, ctypes.c_void_p):
handle = ctypes.c_void_p(handle)
ret_val = ArgVariant() ret_val = ArgVariant()
ret_typeid = ctypes.c_int() ret_typeid = ctypes.c_int()
ret_success = ctypes.c_int() ret_success = ctypes.c_int()
...@@ -35,17 +40,14 @@ def _type_key(handle): ...@@ -35,17 +40,14 @@ def _type_key(handle):
ctypes.byref(ret_val), ctypes.byref(ret_val),
ctypes.byref(ret_typeid), ctypes.byref(ret_typeid),
ctypes.byref(ret_success))) ctypes.byref(ret_success)))
return py_str(ret_val.v_str) return NODE_TYPE.get(py_str(ret_val.v_str), NodeBase)(handle)
NODE_TYPE = {
}
RET_SWITCH = { RET_SWITCH = {
kNull: lambda x: None, kNull: lambda x: None,
kLong: lambda x: x.v_long, kLong: lambda x: x.v_long,
kDouble: lambda x: x.v_double, kDouble: lambda x: x.v_double,
kStr: lambda x: py_str(x.v_str), kStr: lambda x: py_str(x.v_str),
kNodeHandle: lambda x: NODE_TYPE.get(_type_key(x), NodeBase)(x.v_handle) kNodeHandle: lambda x: _return_node(x)
} }
class SliceBase(object): class SliceBase(object):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment