[GitHub] [tvm] junrushao1994 commented on a diff in pull request #11874: [Relay] Handle memory scope during lowering from relay level

[GitBox <gi...@apache.org>]

junrushao1994 commented on code in PR #11874:
URL: https://github.com/apache/tvm/pull/11874#discussion_r909163607


##########
include/tvm/driver/driver_api.h:
##########
@@ -166,6 +166,23 @@ TVM_DLL runtime::Module build(const Map<Target, IRModule>& input, const Target&
  */
 TVM_DLL runtime::Module build(const Map<String, IRModule>& input, const Target& target_host);
 
+/*!
+ * \brief Creates TIR Buffer for provided parameters
+ * \param shape shape of the buffer
+ * \param dtype data type
+ * \param name buffer name
+ * \param data_alignment alignment requirement of data pointer in bytes
+ * \param offset_factor Factor of elem_offset field, elem_offset is guaranteed to be
+ *                      multiple of offset_factor
+                        User can specify data_alignment and offset_factor to be 0
+ *                      A default value will be picked.
+ * \param compact If the statement has already bound to a compact buffer.
+ * \param memory_scope memory scope of the buffer
+ */
+TVM_DLL tir::Buffer BufferWithOffsetAlignment(Array<PrimExpr> shape, DataType dtype,

Review Comment:
   Hey do you think it's possible to move this API to buffer.h?



-- 
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: commits-unsubscribe@tvm.apache.org

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