Browse Source

For aicpu all_shape compile.

tags/v1.2.0
unknown 4 years ago
parent
commit
f640f729c9
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      ge/generator/ge_generator.cc

+ 5
- 1
ge/generator/ge_generator.cc View File

@@ -721,8 +721,12 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &in
GeModelPtr &ge_model = name_to_ge_model.begin()->second;
GELOGD("The opType in op_desc_tmp is [%s]", op_desc_tmp->GetType().c_str());

bool all_shape = false;
(void)AttrUtils::GetBool(op_desc, "_AllShape", all_shape);
bool dynamic_flag = false;
if (CheckShapeReset(op_desc, dynamic_flag) == SUCCESS && dynamic_flag) {
CheckShapeReset(op_desc, dynamic_flag);
if (all_shape && dynamic_flag) {
GELOGD("Get aicpu all_shape kernel!");
vector<GeTensor> inputs_dynamic;
vector<GeTensor> outputs_dynamic;
GE_CHK_STATUS_RET_NOLOG(ResetTensorVecShape(inputs, inputs_dynamic));


Loading…
Cancel
Save