Browse Source

For aicpu all_shape compile.

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

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

@@ -726,7 +726,7 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &in
bool dynamic_flag = false;
(void)AttrUtils::GetBool(op_desc, kAicpuAllshape, all_shape);
CheckShapeReset(op_desc, dynamic_flag);
if (dynamic_flag && all_shape) {
if (dynamic_flag || all_shape) {
GELOGD("Get aicpu all_shape kernel!");
vector<GeTensor> inputs_dynamic;
vector<GeTensor> outputs_dynamic;


Loading…
Cancel
Save