adstraw commented on code in PR #12785: URL: https://github.com/apache/tvm/pull/12785#discussion_r974742950
########## include/tvm/tir/builtin.h: ########## @@ -720,6 +720,16 @@ TVM_DLL const Op& texture2d_load(); */ TVM_DLL const Op& mem_copy(); +/*! + * \brief Initiate a non-blocking DMA copy from source to destination + */ +TVM_DLL const Op& dma_copy(); + +/*! + * \brief Wait until the number of DMAs in flight is less than or equal to some maximum + */ +TVM_DLL const Op& dma_wait(); Review Comment: @masahi what do you think about generalizing that switch by calling it `use_async_copy` and having both the CUDA pass and my new `LowerAsyncDMA` pass use the same switch? -- 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]
