From 71fff26293b7bfc6dc07b30a889e2a9aba590b11 Mon Sep 17 00:00:00 2001
From: l00444296 <lixiwen1@huawei.com>
Date: Thu, 17 Dec 2020 17:30:41 +0800
Subject: [PATCH] Feature: reset shape of dynamic single op

---
 ge/graph/passes/dynamic_single_op_reset_shape_pass.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc b/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc
index 244ba625..3e6377c7 100644
--- a/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc
+++ b/ge/graph/passes/dynamic_single_op_reset_shape_pass.cc
@@ -26,8 +26,8 @@
 namespace ge {
 namespace {
 const int64_t kDynamicShapeDim = -2;
-const char *const kEngineNameAiCpu = "aicpu_ascend_kernel";
-const char *const kEngineNameAiCpuTf = "aicpu_tf_kernel";
+const char *const kEngineNameAiCpu = "DNN_VM_AICPU_ASCEND";
+const char *const kEngineNameAiCpuTf = "DNN_VM_AICPU";
 }  // namespace
 Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) {
   GE_CHECK_NOTNULL(graph);