tqchen commented on code in PR #5:
URL: https://github.com/apache/tvm-ffi/pull/5#discussion_r2347617461
##########
python/tvm_ffi/cython/base.pxi:
##########
@@ -245,6 +246,43 @@ cdef extern from "tvm/ffi/extra/c_env_api.h":
TVMFFIStreamHandle stream,
TVMFFIStreamHandle* opt_out_original_stream)
nogil
+cdef _env_set_current_stream(int device_type, int device_id, uint64_t stream):
Review Comment:
we can expose this as def, so it can be called from python side
##########
python/tvm_ffi/cython/base.pxi:
##########
@@ -236,6 +236,7 @@ cdef extern from "tvm/ffi/c_api.h":
DLTensor* TVMFFITensorGetDLTensorPtr(TVMFFIObjectHandle obj) nogil
DLDevice TVMFFIDLDeviceFromIntPair(int32_t device_type, int32_t device_id)
nogil
+
Review Comment:
new line not needed?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
