Browse Source

feat(lite): add version interface

GitOrigin-RevId: cd09b271a9
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
65c3159cf1
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      src/opr/impl/search_policy/algo_chooser.cpp

+ 3
- 1
src/opr/impl/search_policy/algo_chooser.cpp View File

@@ -414,7 +414,9 @@ size_t AlgoChooser<Opr>::setup_algo(const FixedTensorLayouts& layouts,
size_t workspace = ctx.get_workspace_size_bytes(policy);

std::string ret;
ret.append(mgb_opr->dyn_typeinfo()->name);
if (mgb_opr->dyn_typeinfo()->name) {
ret.append(mgb_opr->dyn_typeinfo()->name);
}
ret += format_fixlayouts<Opr>(layouts, arity_in, arity_out);
Algorithm* palgo = megdnn_opr->get_algorithm_from_desc(policy.algo);
mgb_assert(palgo, "Unknown algo description");


Loading…
Cancel
Save