From 17a37ca7507a9b6923e99771735069cd264aa7cf Mon Sep 17 00:00:00 2001
From: lichun <lichun30@hisilicon.com>
Date: Wed, 23 Jun 2021 14:28:45 +0800
Subject: [PATCH] add atc_params: check_report for ConvertModelToJson

---
 ge/offline/main.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ge/offline/main.cc b/ge/offline/main.cc
index a78ff392..6caed1b7 100755
--- a/ge/offline/main.cc
+++ b/ge/offline/main.cc
@@ -847,6 +847,7 @@ domi::Status GenerateInfershapeJson() {
   ge::Graph graph;
   std::map<string, string> atc_params;
   atc_params.insert(std::pair<string, string>("input_format", FLAGS_input_format));
+  atc_params.insert(std::pair<string, string>("check_report", FLAGS_check_report));
   ret = ParseGraph(graph, atc_params, FLAGS_om.c_str(), FLAGS_weight.c_str(), (domi::FrameworkType) FLAGS_framework,
                    "", FLAGS_target.c_str(), (ge::RunMode) FLAGS_mode, false);
   if (ret != ge::SUCCESS) {