Lunderberg commented on code in PR #12863:
URL: https://github.com/apache/tvm/pull/12863#discussion_r989385552
##########
include/tvm/arith/analyzer.h:
##########
@@ -317,6 +347,82 @@ class CanonicalSimplifier {
Impl* impl_;
};
+/*! \brief Structure for representing result of known
+ *
+ * Values are assigned to allow these flags to be used in bitwise
+ * operations.
+ */
+enum class CompareResult : int {
+ kInconsistent = 0,
+ kEQ = 1,
+ kLT = 2,
+ kLE = 3,
+ kGT = 4,
+ kGE = 5,
+ kNE = 6,
+ kUnknown = 7
+};
+
+inline constexpr CompareResult operator&(CompareResult lhs, CompareResult rhs)
{
+ return CompareResult(static_cast<int>(lhs) & static_cast<int>(rhs));
+}
+inline constexpr CompareResult operator|(CompareResult lhs, CompareResult rhs)
{
+ return CompareResult(static_cast<int>(lhs) | static_cast<int>(rhs));
+}
+
+/*!
+ * \brief Using previously specified knowns, compare the expressions provided
+ *
+ * Given known expressions [(a OP b), (b OP c), ..., (y OP z)], search
+ * for a known result for `(a OP z)`.
+ */
+class TransitiveComparisonAnalyzer {
+ public:
+ /* \brief Using previously specified knowns, compare the expressions provided
+ *
+ * \param lhs The left-hand side of the comparison
+ *
+ * \param rhs The right-hand side of the comparison
+ *
+ * \return The most specific result that can be proven about the
+ * comparison. If nothing can be proven, returns kUnknown.
+ */
+ CompareResult TryCompare(const PrimExpr& lhs, const PrimExpr& rhs);
+
+ /*! \brief Bind a variable as being equal to a known expression
+ *
+ * \param var The variable of interest.
+ * \param expr The bound expression
+ * \param allow_override Whether to allow override of existing information.
+ */
+ void Bind(const Var& var, const PrimExpr& expr, bool allow_override = false);
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: commits-unsubscr...@tvm.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
