Fix bug of aicpu all_shape compile.

4 years ago · 4b421a4c18
parent edd784700f
commit 4b421a4c18
2 changed files with 2 additions and 2 deletions
--- a/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc
+++ b/tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc
@ -140,7 +140,7 @@ TEST_F(UtestKernelExTaskInfo, kernel_ex_task_info_calculate_args) {
 TEST_F(UtestKernelExTaskInfo, kernel_ex_task_ext_info) {
  const string ext_info = {1, 1, 1, 1, 0, 0, 0, 0};
  const OpDescPtr op_desc = CreateOpDesc("FrameworkOp", "FrameworkOp");
-  AttrUtils::SetBool(op_desc, "_Allshape", true);
+  AttrUtils::SetBool(op_desc, "_AllShape", true);

  KernelExTaskInfo kernel_ex_task_info;
  EXPECT_EQ(kernel_ex_task_info.InitTaskExtInfo(ext_info, op_desc), SUCCESS);
--- a/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
+++ b/tests/ut/ge/graph/load/kernel_task_info_unittest.cc
@ -390,7 +390,7 @@ TEST_F(UtestKernelTaskInfo, init_kernel_taskInfo_with_aicpu_kernel_type_fail) {
  rtStreamCreate(&stream, 0);
  model.stream_list_ = { stream };
  model.op_list_[0] = CreateOpDesc("FrameworkOp", "FrameworkOp");
-  AttrUtils::SetBool(model.op_list_[0], "_Allshape", true);
+  AttrUtils::SetBool(model.op_list_[0], "_AllShape", true);

  domi::TaskDef task_def;
  KernelTaskInfo kernel_task_info;