diff --git a/tf_adapter/kernels/aicpu/dataset_function.cc b/tf_adapter/kernels/aicpu/dataset_function.cc index e0bf7903bf6c04466ea6fe886f304add78148185..989263cf353bc32e3f7795296063233c6c0222e4 100644 --- a/tf_adapter/kernels/aicpu/dataset_function.cc +++ b/tf_adapter/kernels/aicpu/dataset_function.cc @@ -613,6 +613,7 @@ Status DatasetFunction::Instantialte() { std::map graph_options; // add graph level tag to exclude aicore engine graph_options[ge::AscendString("ge.exec.exclude_engines")] = ge::AscendString("AiCore"); + graph_options[ge::AscendString("ge.exec.graphExecTimeout")] = ge::AscendString("60"); auto ret_build = aclgrphBuildModel(ge_graph_, graph_options, ge_model_); DATASET_REQUIRES(ret_build == ge::GRAPH_SUCCESS, errors::Unavailable("Build ge model failed.", ret_build));