提交 3b28af95 编写于 作者: M Megvii Engine Team

fix(sdk): add disable-optimize-for-inference option

GitOrigin-RevId: 7e6ea13be7eaa70cd0d49f20571c4014d3a20e07
上级 938152af
...@@ -17,11 +17,11 @@ import numpy as np ...@@ -17,11 +17,11 @@ import numpy as np
import megengine as mge import megengine as mge
import megengine.core._imperative_rt as rt import megengine.core._imperative_rt as rt
import megengine.core.tensor.megbrain_graph as G import megengine.core.tensor.megbrain_graph as G
from megengine.utils import comp_graph_tools as cgtools from megengine import tensor
from megengine.core.ops import builtin
from megengine.core._imperative_rt.core2 import apply from megengine.core._imperative_rt.core2 import apply
from megengine.core.ops import builtin
from megengine.core.tensor.megbrain_graph import VarNode from megengine.core.tensor.megbrain_graph import VarNode
from megengine import tensor from megengine.utils import comp_graph_tools as cgtools
logger = mge.get_logger(__name__) logger = mge.get_logger(__name__)
...@@ -372,8 +372,8 @@ def main(): ...@@ -372,8 +372,8 @@ def main():
) )
parser.add_argument( parser.add_argument(
"--optimize-for-inference", "--optimize-for-inference",
action="store_false", action="store_true",
help="enbale optimization for inference", help="enable optimization for inference",
) )
parser.add_argument( parser.add_argument(
"--no-assert", "--no-assert",
...@@ -498,7 +498,7 @@ def main(): ...@@ -498,7 +498,7 @@ def main():
wrap_output_vars, wrap_output_vars,
append_json=True, append_json=True,
strip_info_file=strip_info_file, strip_info_file=strip_info_file,
**sereg_kwargs **sereg_kwargs,
) )
fout.write(dump_content) fout.write(dump_content)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册