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);