Browse Source

Fix bug of aicpu all_shape compile.

tags/v1.2.0
unknown 4 years ago
parent
commit
4b421a4c18
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc
  2. +1
    -1
      tests/ut/ge/graph/load/kernel_task_info_unittest.cc

+ 1
- 1
tests/ut/ge/graph/load/kernel_ex_task_info_unittest.cc View File

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


+ 1
- 1
tests/ut/ge/graph/load/kernel_task_info_unittest.cc View File

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


Loading…
Cancel
Save