From 3b28af958a9ffb13df3f40a39515a55525d05101 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 25 Jan 2021 20:31:46 +0800 Subject: [PATCH] fix(sdk): add disable-optimize-for-inference option GitOrigin-RevId: 7e6ea13be7eaa70cd0d49f20571c4014d3a20e07 --- sdk/load-and-run/dump_with_testcase_mge.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/sdk/load-and-run/dump_with_testcase_mge.py b/sdk/load-and-run/dump_with_testcase_mge.py index 65c013461..02010e750 100755 --- a/sdk/load-and-run/dump_with_testcase_mge.py +++ b/sdk/load-and-run/dump_with_testcase_mge.py @@ -17,11 +17,11 @@ import numpy as np import megengine as mge import megengine.core._imperative_rt as rt import megengine.core.tensor.megbrain_graph as G -from megengine.utils import comp_graph_tools as cgtools -from megengine.core.ops import builtin +from megengine import tensor from megengine.core._imperative_rt.core2 import apply +from megengine.core.ops import builtin 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__) @@ -372,8 +372,8 @@ def main(): ) parser.add_argument( "--optimize-for-inference", - action="store_false", - help="enbale optimization for inference", + action="store_true", + help="enable optimization for inference", ) parser.add_argument( "--no-assert", @@ -498,7 +498,7 @@ def main(): wrap_output_vars, append_json=True, strip_info_file=strip_info_file, - **sereg_kwargs + **sereg_kwargs, ) fout.write(dump_content) -- GitLab