Browse Source

fix(jit/cpu): make lar -enable_jit args possible

on other backends, for example cpu with MLIR

GitOrigin-RevId: ef98d46be3
master
Megvii Engine Team 2 years ago
parent
commit
18d7a97c68
2 changed files with 0 additions and 6 deletions
  1. +0
    -5
      lite/src/mge/network_impl.cpp
  2. +0
    -1
      src/core/include/megbrain/graph/cg.h

+ 0
- 5
lite/src/mge/network_impl.cpp View File

@@ -79,11 +79,6 @@ void NetworkImplDft::application_config() {
force_output_use_user_specified_memory,
force_output_use_user_specified_memory);
ConfigOption(no_profiling_on_shape_change, no_profiling_on_shape_change);
LITE_ASSERT(
m_user_config->options.jit_level == 0 ||
(m_user_config->options.jit_level > 0 &&
device_type == LiteDeviceType::LITE_CUDA),
"jit only support in cuda device.");
ConfigOption(graph_opt.jit, jit_level);
ConfigOption(comp_node_seq_record_level, comp_node_seq_record_level);
ConfigOption(graph_opt_level, graph_opt_level);


+ 0
- 1
src/core/include/megbrain/graph/cg.h View File

@@ -366,7 +366,6 @@ public:
* construction
* 2: level-2: level-1, plus global optimization before graph
* compiling
* 3: also enable JIT
* <0: corresponding level, with result check for debug
*/
int16_t graph_opt_level = 2;


Loading…
Cancel
Save