提交 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
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/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;
});
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;
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);
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) {
#if !MGB_ENABLE_FASTRUN
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");
}
#endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册