[Relay] Fix name conflict in PartialEval (apache#3402)

neo-ai · Jun 27, 2019 · e629e23 · e629e23
1 parent d6e335a
commit e629e23
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/relay/pass/partial_eval.cc b/src/relay/pass/partial_eval.cc
@@ -101,6 +101,7 @@
 
 namespace tvm {
 namespace relay {
+namespace partial_eval {
 
 using namespace runtime;
 
@@ -1087,12 +1088,14 @@ Expr PostProcess(const Expr& e) {
   return StripWithFuncId(DeDup(Remap(e)));
 }
 
+}  // namespace partial_eval
+
 Expr PartialEval(const Expr& e, const Module& m) {
   return TransformF([&](const Expr& e) {
       return LetList::With([&](LetList* ll) {
-          PartialEvaluator pe(FreeVars(e), m);
+          relay::partial_eval::PartialEvaluator pe(FreeVars(e), m);
           pe.InitializeFuncId(e);
-          return PostProcess(pe.VisitExpr(e, ll)->dynamic);
+          return relay::partial_eval::PostProcess(pe.VisitExpr(e, ll)->dynamic);
         });
     }, e);
 }