From 3f69c0ca6515544dfa69fe6251f3c53a7ad8ce14 Mon Sep 17 00:00:00 2001 From: unknown Date: Mon, 25 Jan 2021 22:02:49 +0800 Subject: [PATCH] For aicpu all_shape compile. --- ge/generator/ge_generator.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc index 52dfa65d..6c0fb6e5 100644 --- a/ge/generator/ge_generator.cc +++ b/ge/generator/ge_generator.cc @@ -726,7 +726,7 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector &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 inputs_dynamic; vector outputs_dynamic;