Is there a ready-made function in tvm to realize follow function:
Two PrimExpr dose have same vars, eg:
a: 2 * j +3 + 5 * i
b: 15 - j - i
To find i is in both exprs
Is there a ready-made function in tvm to realize follow function:
Two PrimExpr dose have same vars, eg:
a: 2 * j +3 + 5 * i
b: 15 - j - i
To find i is in both exprs
Try using ExprUseVar
at include/tvm/tir/analysis.h
/*!
* \brief Whether e expression used any var in variable set..
* \param expr The expression to be checked.
* \param vset_contains The check function to see if var is in the vset.
* \return Whether e uses vset.
*/
TVM_DLL bool ExprUseVar(const PrimExpr& expr, std::function<bool(const VarNode*)> vset_contains);
OK, Get it. Thanks very much