Lunderberg commented on a change in pull request #9727:
URL: https://github.com/apache/tvm/pull/9727#discussion_r807365129
##########
File path: include/tvm/tir/function.h
##########
@@ -136,15 +157,31 @@ class PrimFunc : public BaseFunc {
public:
/*!
* \brief Constructor
+ *
* \param params The parameters of the function.
+ *
* \param body The body of the function.
+ *
* \param ret_type The return type of the function.
+ *
* \param buffer_map The buffer map for parameter buffer unpacking.
+ * This contains buffer objects as they appear in the body of the
+ * PrimFunc. (e.g. a buffer of shape ``[1024]`` originally
+ * generated as a tensor of shape ``[32, 32]``)
+ *
+ * \param preflattened_buffer_map The buffer map for
+ * parameter buffer unpacking. This contains buffer
+ * objects as they are expected to be passed in by the
+ * callee. (e.g. a buffer of shape ``[32, 32]`` originally
+ * generated as a tensor of shape ``[32, 32]``)
+ *
* \param attrs Additional function attributes.
+ *
* \param span The location of this object in the source code.
*/
TVM_DLL PrimFunc(Array<tir::Var> params, Stmt body, Type ret_type =
VoidType(),
Map<tir::Var, Buffer> buffer_map = Map<tir::Var, Buffer>(),
+ Map<tir::Var, Buffer> preflattened_buffer_map =
Map<tir::Var, Buffer>(),
Review comment:
And updated.
--
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]
