diff --git a/tf_adapter/interface_spec/api_npu_config.pyh b/tf_adapter/interface_spec/api_npu_config.pyh index f9f1a9b02917f9da17f205cac67615d86db377a9..a693c42f1d78936270fa5a585e8cc218cb3e9a63 100644 --- a/tf_adapter/interface_spec/api_npu_config.pyh +++ b/tf_adapter/interface_spec/api_npu_config.pyh @@ -18,7 +18,7 @@ class NPURunConfig(run_config_lib.RunConfig): distribute_config=None, modify_mixlist=None, op_precision_mode=None, device_type="default_device_type", soc_config=None, hccl_timeout=None, op_wait_timeout=None, op_execute_timeout=None, HCCL_algorithm=None, customize_dtypes=None, op_debug_config=None, memory_config=None, experimental_config=None, - jit_compile=True, topo_sorting_mode=None, aoe_config_file=None, insert_op_file=None): + jit_compile=False, topo_sorting_mode=None, aoe_config_file=None, insert_op_file=None): class ProfilingConfig(): def __init__(self, enable_profiling=False, profiling_options=None): diff --git a/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py b/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py index 721ec8eb1f2d12de05253aa06a5814d38b72eaa2..afbb1073d4848f21122b316d687d29718e306d84 100644 --- a/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py +++ b/tf_adapter/python/npu_bridge/estimator/npu/npu_config.py @@ -100,7 +100,7 @@ class NPURunConfig(run_config_lib.RunConfig): op_debug_config=None, memory_config=None, experimental_config=None, - jit_compile=True, + jit_compile=False, topo_sorting_mode=None, aoe_config_file=None, insert_op_file=None diff --git a/tf_adapter/util/npu_attrs.cc b/tf_adapter/util/npu_attrs.cc index 1aba4c18323602ea152fc8c3c2749939c73a5762..12fbaf9509b3209d83b1d52b7796c304ac5e75a9 100644 --- a/tf_adapter/util/npu_attrs.cc +++ b/tf_adapter/util/npu_attrs.cc @@ -392,7 +392,7 @@ std::map NpuAttrs::GetSessOptions(const OpKernelConstr std::string HCCL_algorithm; std::string atomic_clean_policy = "0"; std::string static_memory_policy; - std::string jit_compile = "1"; + std::string jit_compile = "0"; std::string topo_sorting_mode; std::string insert_op_file; std::string resource_config_path; @@ -1002,7 +1002,7 @@ std::map NpuAttrs::GetAllAttrOptions(const AttrSlice & std::string graph_exec_timeout; std::string logical_device_cluster_deploy_mode = "LB"; std::string logical_device_id; - std::string jit_compile = "1"; + std::string jit_compile = "0"; std::string topo_sorting_mode; std::string insert_op_file; std::string resource_config_path; @@ -1512,7 +1512,7 @@ Status NpuAttrs::SetNpuOptimizerAttr(const GraphOptimizationPassOptions &options int64_t graph_exec_timeout = 600000L; std::string logical_device_cluster_deploy_mode = "LB"; std::string logical_device_id; - bool jit_compile = true; + bool jit_compile = false; std::string aoe_config_file; const RewriterConfig &rewrite_options = diff --git a/tf_adapter_2.x/python/npu_device/configs/npu_config.py b/tf_adapter_2.x/python/npu_device/configs/npu_config.py index b81be133bf67c3f9bc648693d52b445b8f3bf64c..1259e93025bd18c6bc42a2c163e8cb1fa10a1b11 100644 --- a/tf_adapter_2.x/python/npu_device/configs/npu_config.py +++ b/tf_adapter_2.x/python/npu_device/configs/npu_config.py @@ -55,7 +55,7 @@ class NpuConfig(NpuBaseConfig): self.profiling_config = NpuProfilingConfig() self.enable_small_channel = OptionValue(False, [True, False]) self.graph_exec_timeout = OptionValue(None, None) - self.jit_compile = OptionValue(True, [True, False]) + self.jit_compile = OptionValue(False, [True, False]) self.topo_sorting_mode = OptionValue(None, [0, 1, None]) self.customize_dtypes = OptionValue(None, None) self.overflow_flag = OptionValue(1, [0, 1])