Skip to content

Commit 3c0c18b

Browse files
fix rebase problem
1 parent 40032dd commit 3c0c18b

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

include/tvm/relay/pass.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -399,11 +399,12 @@ TVM_DLL Expr ToGraphNormalForm(const Expr& e);
399399
* It has two benefit: remove runtime overhead, and allow more optimization (typically fusion).
400400
* As a side effect, code size will explode.
401401
*
402-
* \param e the expression,
402+
* \param e the expression
403+
* \param mod the module
403404
*
404405
* \return the optimized expression.
405406
*/
406-
TVM_DLL Expr PartialEval(const Expr& e);
407+
TVM_DLL Expr PartialEval(const Expr& e, const Module& mod);
407408

408409
/*! \brief A hashing structure in the style of std::hash. */
409410
struct StructuralHash {

src/relay/pass/partial_eval.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1101,7 +1101,7 @@ namespace transform {
11011101
Pass PartialEval() {
11021102
runtime::TypedPackedFunc<Function(Function, Module, PassContext)> pass_func =
11031103
[=](Function f, Module m, PassContext pc) {
1104-
return Downcast<Function>(PartialEval(f));
1104+
return Downcast<Function>(PartialEval(f, m));
11051105
};
11061106
return CreateFunctionPass(pass_func, 1, "partial_eval", {});
11071107
}

0 commit comments

Comments
 (0)