Browse Source

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

GitOrigin-RevId: e58caf08c8
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
9da26407ea
3 changed files with 9 additions and 8 deletions
  1. +1
    -0
      .gitattributes
  2. +7
    -7
      imperative/python/src/graph_rt.cpp
  3. +1
    -1
      src/gopt/impl/inference.cpp

+ 1
- 0
.gitattributes View File

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

+ 7
- 7
imperative/python/src/graph_rt.cpp View File

@@ -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!");


+ 1
- 1
src/gopt/impl/inference.cpp View File

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


Loading…
Cancel
Save