Fix bug of aicpu all_shape compile.

pull/1063/head
unknown 4 years ago
parent b9fd09661e
commit edd784700f

@ -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");
(void)AttrUtils::GetBool(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);

@ -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");
(void)AttrUtils::GetBool(op_list_[0], "_Allshape", true);
AttrUtils::SetBool(model.op_list_[0], "_Allshape", true);
domi::TaskDef task_def;
KernelTaskInfo kernel_task_info;

Loading…
Cancel
Save