Browse Source

fix(lite): fix BUILD and misc

GitOrigin-RevId: 48aa0e0924
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
3fab21ed31
2 changed files with 2 additions and 4 deletions
  1. +1
    -1
      src/opr/impl/dnn/dnn.sereg.h
  2. +1
    -3
      src/opr/impl/search_policy/algo_chooser.cpp

+ 1
- 1
src/opr/impl/dnn/dnn.sereg.h View File

@@ -183,7 +183,7 @@ struct ConvLoadDumpImpl {
static void dump(OprDumpContext& ctx, const cg::OperatorNodeBase& opr_) {
auto&& opr = opr_.cast_final_safe<Opr>();
ctx.write_param<ConvParam>(opr.param());
ctx.write_param<megdnn::param::ExecutionPolicy>(opr.execution_policy());
ctx.write_param<megdnn::param::ExecutionPolicy>(opr.execution_policy_transient());
}

static VarNode* make(const cg::VarNodeArray& inputs, const ConvParam& param,


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

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

std::string ret;
if (mgb_opr->dyn_typeinfo()->name) {
ret.append(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