提交 9da26407 编写于 作者: M Megvii Engine Team

feat(ci): and model to test algo policy compatible

GitOrigin-RevId: e58caf08c8488cf2416b24c9deb5a7a50dab89bf
上级 e19b9af1
...@@ -12,3 +12,4 @@ ci/resource/models/float/mobilenet_v2.pkl filter=lfs diff=lfs merge=lfs -text ...@@ -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) { ...@@ -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{ const std::unordered_map<std::string, std::function<void()>> m{
{"HEURISTIC", [&](){ stg = _AlgoStrategy::HEURISTIC; }}, {"HEURISTIC", [&]() { stg = _AlgoStrategy::HEURISTIC; }},
{"HEURISTIC_REPRODUCIBLE", [&](){ stg = _AlgoStrategy::HEURISTIC_REPRODUCIBLE; }}, {"PROFILE", [&]() { stg = _AlgoStrategy::PROFILE; }},
{"PROFILE", [&](){ stg = _AlgoStrategy::PROFILE; }}, {"REPRODUCIBLE", [&]() { stg = _AlgoStrategy::REPRODUCIBLE; }},
{"PROFILE_REPRODUCIBLE", [&](){ stg = _AlgoStrategy::PROFILE_REPRODUCIBLE; }}, {"OPTMIZED", [&]() { stg = _AlgoStrategy::OPTMIZED; }},
{"PROFILE_HEURISTIC", [&](){ stg = _AlgoStrategy::PROFILE_HEURISTIC; }},
}; };
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( ...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册