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 | ||||