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 0d24e980a1c277ab9692917ed9db1d1502bd5d30..9ed2c07688433dcbdf74e766ef36a27399f84ecd 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 @@ -60,7 +60,7 @@ class NpuConfig(NpuBaseConfig): self.deterministic = OptionValue(0, [0, 1]) self.op_precision_mode = OptionValue(None, None) self.graph_exec_timeout = OptionValue(None, None) - self.topo_sorting_mode = OptionValue(None, [0, 1, None]) + self.topo_sorting_mode = OptionValue(None, [0, 1, 2, None]) self.customize_dtypes = OptionValue(None, None) self.overflow_flag = OptionValue(1, [0, 1]) self.stream_sync_timeout = OptionValue(-1, None)