junrushao1994 commented on a change in pull request #8693:
URL: https://github.com/apache/tvm/pull/8693#discussion_r686491273
##########
File path: src/tir/schedule/analysis.h
##########
@@ -224,6 +237,17 @@ std::vector<TypedPackedFunc<CommReducer(DataType)>>
GetReducerGetters();
bool FromIdentityCombiner(const PrimExpr& identity, const BufferStore&
combiner,
CommReducer* result_reducer, PrimExpr* lhs,
PrimExpr* rhs);
+/******** Annotation ********/
+
+/*!
+ * \brief Create a new block with the given annotation added
+ * \param block The block with original annotation
+ * \param attr_key The annotation key to be added
+ * \param attr_value The annotation value to be added
+ * \return A new block with the given annotation as its last annotation
+ */
+Block WithAnnotation(const BlockNode* block, const String& attr_key, const
ObjectRef& attr_value);
Review comment:
Move this to a new file `transform.h` given it is not analysis
--
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]
