GitOrigin-RevId: e58caf08c8
tags/v1.3.0
| @@ -12,3 +12,4 @@ ci/resource/models/float/mobilenet_v2.pkl filter=lfs diff=lfs merge=lfs -text | |||||
| ci/resource/models/float/shufflenet_v2.pkl filter=lfs diff=lfs merge=lfs -text | ci/resource/models/float/shufflenet_v2.pkl filter=lfs diff=lfs merge=lfs -text | ||||
| ci/resource/dump/roi_align_backward_8.8.0.mdl filter=lfs diff=lfs merge=lfs -text | ci/resource/dump/roi_align_backward_8.8.0.mdl filter=lfs diff=lfs merge=lfs -text | ||||
| ci/resource/dump/relayout_format_8.10.0.mdl filter=lfs diff=lfs merge=lfs -text | ci/resource/dump/relayout_format_8.10.0.mdl filter=lfs diff=lfs merge=lfs -text | ||||
| ci/resource/dump/batch_conv_bias_with_policy_8.8.0.mdl filter=lfs diff=lfs merge=lfs -text | |||||
| @@ -259,14 +259,14 @@ void init_graph_rt(py::module m) { | |||||
| return vars; | return vars; | ||||
| }); | }); | ||||
| m.def("modify_opr_algo_strategy_inplace", [](const VarNodeArray& dest_vars, const std::string& strategy) { | |||||
| m.def("modify_opr_algo_strategy_inplace", [](const VarNodeArray& dest_vars, | |||||
| const std::string& strategy) { | |||||
| _AlgoStrategy stg; | _AlgoStrategy stg; | ||||
| const std::unordered_map<std::string,std::function<void()>> m{ | |||||
| {"HEURISTIC", [&](){ stg = _AlgoStrategy::HEURISTIC; }}, | |||||
| {"HEURISTIC_REPRODUCIBLE", [&](){ stg = _AlgoStrategy::HEURISTIC_REPRODUCIBLE; }}, | |||||
| {"PROFILE", [&](){ stg = _AlgoStrategy::PROFILE; }}, | |||||
| {"PROFILE_REPRODUCIBLE", [&](){ stg = _AlgoStrategy::PROFILE_REPRODUCIBLE; }}, | |||||
| {"PROFILE_HEURISTIC", [&](){ stg = _AlgoStrategy::PROFILE_HEURISTIC; }}, | |||||
| const std::unordered_map<std::string, std::function<void()>> m{ | |||||
| {"HEURISTIC", [&]() { stg = _AlgoStrategy::HEURISTIC; }}, | |||||
| {"PROFILE", [&]() { stg = _AlgoStrategy::PROFILE; }}, | |||||
| {"REPRODUCIBLE", [&]() { stg = _AlgoStrategy::REPRODUCIBLE; }}, | |||||
| {"OPTMIZED", [&]() { stg = _AlgoStrategy::OPTMIZED; }}, | |||||
| }; | }; | ||||
| auto it = m.find(strategy); | auto it = m.find(strategy); | ||||
| mgb_assert(it != m.end(), "Invalid strategy string!"); | mgb_assert(it != m.end(), "Invalid strategy string!"); | ||||
| @@ -154,7 +154,7 @@ void gopt::modify_opr_algo_strategy_inplace( | |||||
| opr::mixin::AlgoChooserHelper::ExecutionPolicy::Strategy strategy) { | opr::mixin::AlgoChooserHelper::ExecutionPolicy::Strategy strategy) { | ||||
| #if !MGB_ENABLE_FASTRUN | #if !MGB_ENABLE_FASTRUN | ||||
| using S = opr::mixin::AlgoChooserHelper::ExecutionPolicy::Strategy; | using S = opr::mixin::AlgoChooserHelper::ExecutionPolicy::Strategy; | ||||
| if (strategy == S::PROFILE || strategy == S::PROFILE_REPRODUCIBLE) { | |||||
| if ((strategy & S::PROFILE) && !(strategy & S::HEURISTIC)) { | |||||
| mgb_throw(MegBrainError, "fastrun is disabled at compile time"); | mgb_throw(MegBrainError, "fastrun is disabled at compile time"); | ||||
| } | } | ||||
| #endif | #endif | ||||