You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/03/03 16:13:27 UTC

[GitHub] [tvm] tqchen commented on a change in pull request #7573: [Runtime] Extend Graph Runtime To Support Cuda Graph Launch

tqchen commented on a change in pull request #7573:
URL: https://github.com/apache/tvm/pull/7573#discussion_r586551771



##########
File path: include/tvm/runtime/c_runtime_api.h
##########
@@ -481,6 +481,15 @@ TVM_DLL int TVMStreamFree(int device_type, int device_id, TVMStreamHandle stream
  */
 TVM_DLL int TVMSetStream(int device_type, int device_id, TVMStreamHandle handle);
 
+
+TVM_DLL int TVMStreamBeginCapture(int device_type, int device_id, TVMStreamHandle stream);

Review comment:
       Given cuGraph is specific to CUDA atm, let us not introduce it to the DeviceAPI for now, instead just use the cuda api in cugraph runtime




----------------------------------------------------------------
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.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org