Browse Source

perf(subgraph): remove unnecessary sync

GitOrigin-RevId: 1546286706
tags/v1.6.0
Megvii Engine Team 4 years ago
parent
commit
ca4374b6e2
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      imperative/src/impl/ops/utility.cpp

+ 1
- 0
imperative/src/impl/ops/utility.cpp View File

@@ -321,6 +321,7 @@ ComputingGraphHolder& get_computing_graph(std::shared_ptr<OpDef> compiled_op, Sm
cg_holder.graph->options().async_exec_level = 0;
cg_holder.graph->options().graph_opt_level = compiled_op->cast_final_safe<CompiledOp>().gopt_level;
cg_holder.graph->options().enable_var_mem_defragment = false;
cg_holder.graph->options().comp_seq_sync_device = false;
// set allocator for DTR support
cg_holder.graph->set_device_memory_allocator(cg_holder.allocator);
VarNodeArray input_vars;


Loading…
Cancel
Save