From 18d7a97c687d8f324458bb6345b64bed9af7c620 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 7 Nov 2022 15:05:20 +0800 Subject: [PATCH] fix(jit/cpu): make lar -enable_jit args possible on other backends, for example cpu with MLIR GitOrigin-RevId: ef98d46be326417c85fcdc482f02758e352e2f52 --- lite/src/mge/network_impl.cpp | 5 ----- src/core/include/megbrain/graph/cg.h | 1 - 2 files changed, 6 deletions(-) diff --git a/lite/src/mge/network_impl.cpp b/lite/src/mge/network_impl.cpp index 10480c44c..a77195206 100644 --- a/lite/src/mge/network_impl.cpp +++ b/lite/src/mge/network_impl.cpp @@ -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); diff --git a/src/core/include/megbrain/graph/cg.h b/src/core/include/megbrain/graph/cg.h index 2d545bcb9..0ae7a2365 100644 --- a/src/core/include/megbrain/graph/cg.h +++ b/src/core/include/megbrain/graph/cg.h @@ -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; -- GitLab