Browse Source

fixv ut

pull/2039/head
guopeian 3 years ago
parent
commit
96853e7c3f
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/single_op/task/op_task.cc
  2. +1
    -1
      tests/ut/ge/single_op/single_op_model_unittest.cc

+ 1
- 1
ge/single_op/task/op_task.cc View File

@@ -585,7 +585,7 @@ Status AiCpuBaseTask::SetExtInfoAndType(const std::string &kernel_ext_info, uint

int32_t unknown_shape_type_val = 0;
(void) AttrUtils::GetInt(op_desc_, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, unknown_shape_type_val);
GELOGD("Get unknown_type is %d.", unknown_shape_type_val);
GELOGI("Get unknown_type is %d.", unknown_shape_type_val);
unknown_type_ = static_cast<UnknowShapeOpType>(unknown_shape_type_val);

aicpu_ext_handle_.reset(new(std::nothrow) ::ge::hybrid::AicpuExtInfoHandler(op_desc_->GetName(),


+ 1
- 1
tests/ut/ge/single_op/single_op_model_unittest.cc View File

@@ -409,7 +409,7 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) {
DynamicSingleOp single_op(0, &stream_mu, stream);
model.model_helper_.model_ = ge_model;
auto op_desc = std::make_shared<ge::OpDesc>("add", "Add");
AttrUtils::SetInt(op_desc, ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE);
AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE);
NodePtr node = graph->AddNode(op_desc);
model.op_list_[0] = node;
StreamResource *res = new (std::nothrow) StreamResource(1);


Loading…
Cancel
Save