Lunderberg commented on code in PR #11970:
URL: https://github.com/apache/tvm/pull/11970#discussion_r920061151
##########
include/tvm/arith/analyzer.h:
##########
@@ -365,6 +365,36 @@ class IntSetAnalyzer {
*/
TVM_DLL IntSet operator()(const PrimExpr& expr, const Map<Var, IntSet>&
dom_map);
+ /*!
+ * \brief Find a symbolic integer set that contains all possible
+ * values of expr given the domain of each variables, using
+ * the domain map defined by bound variables.
+ *
+ * \param expr The expression of interest.
+ * \return the result of the analysis.
+ */
+ TVM_DLL IntSet operator()(const PrimExpr& expr);
+
+ /*!
+ * \brief Update binding of var to a new expression.
+ *
+ * \param var The variable of interest.
+ * \param new_interval_set The set of allowed values for this var.
+ * \param allow_override whether we allow override of existing information.
+ */
+ TVM_DLL void Update(const Var& var, const IntSet& new_interval_set, bool
allow_override = false);
+
+ /*!
+ * \brief Update binding of var to a new expression.
+ *
+ * \param var The variable of interest.
+ * \param new_range The range of allowed values for this var.
+ * \param allow_override whether we allow override of existing information.
+ */
+ TVM_DLL void Update(const Var& var, const Range& new_range, bool
allow_override = false);
+
+ std::function<void()> EnterConstraint(const PrimExpr& constraint);
Review Comment:
Thank you, and added.
--
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]
