From 775d53be2ced3ae0cee4492f703070efd10bcd8c Mon Sep 17 00:00:00 2001
From: y00500818 <yangyongqiang7@hisilicon.com>
Date: Thu, 13 May 2021 10:55:47 +0800
Subject: [PATCH] add infer format for single op

---
 ge/generator/ge_generator.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc
index a12f3cf7..fa6d8fa8 100644
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@@ -843,6 +843,9 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &in
   Graph graph;
   GE_CHK_STATUS(BuildSingleOpGraph(op_desc, inputs, outputs, name, graph),
                 "[Build][Graph] for single op:%s fail.", op_desc->GetName().c_str());
+  auto op = OpDescUtils::CreateOperatorFromOpDesc(op_desc);
+  GE_CHK_STATUS_RET(op_desc->CallInferFormatFunc(op),
+                    "[Call][InferFormatFunc] for single op:%s fail.", op_desc->GetName().c_str());
 
   // 2. check engine type when compile online
   if (model_file_name == kFileNameSuffix) {