From b6f9b3896abae422a93f77190c52b61d13d0ce8e Mon Sep 17 00:00:00 2001 From: yaolun Date: Mon, 14 Nov 2022 09:58:40 +0800 Subject: [PATCH] Add timeout settings for acl graph. --- tf_adapter/kernels/aicpu/dataset_function.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tf_adapter/kernels/aicpu/dataset_function.cc b/tf_adapter/kernels/aicpu/dataset_function.cc index e0bf7903b..989263cf3 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)); -- Gitee