For aicpu all_shape compile.

4 years ago · 3f69c0ca65
parent d383e91c52
commit 3f69c0ca65
1 changed files with 1 additions and 1 deletions
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@ -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;