Skip to content

Commit

Permalink
committed fix
Browse files Browse the repository at this point in the history
  • Loading branch information
tqchen committed Jan 28, 2017
1 parent adff6f9 commit a2cf082
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions python/tvm/_ctypes/_function.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@ def cfun(args, type_codes, num_args, ret, _):
ret = TVMRetValueHandle(ret)
check_call(_LIB.TVMCFuncSetReturn(ret, values[0], ctypes.c_int(tcodes[0])))
_ = temp_args
_ = rv

handle = FunctionHandle()
f = TVMPackedCFunc(cfun)
Expand Down Expand Up @@ -143,6 +144,7 @@ def __call__(self, *args):
self.handle, values, tcodes, ctypes.c_int(num_args),
ctypes.byref(ret_val), ctypes.byref(ret_tcode)))
_ = temp_args
_ = args
return RETURN_SWITCH[ret_tcode.value](ret_val)


Expand Down

0 comments on commit a2cf082

Please sign in to comment.