diff --git a/src/relay/pass/legalize.cc b/src/relay/pass/legalize.cc index f57d9103412e..daed47afdcb3 100644 --- a/src/relay/pass/legalize.cc +++ b/src/relay/pass/legalize.cc @@ -103,7 +103,7 @@ Pass Legalize(const std::string& legalize_map_attr_name) { [=](Function f, Module m, PassContext pc) { return Downcast(relay::legalize::Legalize(f, legalize_map_attr_name)); }; - return CreateFunctionPass(pass_func, 0, "Legalize", {ir::StringImm::make("InferType")}); + return CreateFunctionPass(pass_func, 1, "Legalize", {ir::StringImm::make("InferType")}); } TVM_REGISTER_API("relay._transform.Legalize").set_body_typed(Legalize);