lhutton1 commented on a change in pull request #9649:
URL: https://github.com/apache/tvm/pull/9649#discussion_r763084001
##########
File path: include/tvm/tir/usmp/utils.h
##########
@@ -153,6 +153,45 @@ class BufferInfo : public ObjectRef {
TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS(BufferInfo, ObjectRef, BufferInfoNode);
};
+/*!
+ * \brief This is a composite node that is produced by extract_buffer_info
+ * analysis pass that contains useful global information that could be useful
+ * for memory planning algorithms.
+ */
+struct BufferInfoAnalysisNode : public Object {
+ /*! \brief The BufferInfo object and its associated TIR statement */
+ Map<BufferInfo, tir::Stmt> buffer_info_stmts;
+ /*! \brief This represent maximum amount of memory being used at
+ * any point of time in the inference. This value is largely the
+ * best allocation an algorithm could achieve. Due to
Review comment:
That's great, thanks!
--
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]
