From ce47c5c0f90a55d07fa8cd7f542a976fc9d42f27 Mon Sep 17 00:00:00 2001
From: WongGawa <haungjiahua6@huawei.com>
Date: Tue, 27 Aug 2024 15:00:46 +0800
Subject: [PATCH] fix yolov7's MaxPool2d's bug

---
 demo/predict.py | 7 +++++++
 test.py         | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/demo/predict.py b/demo/predict.py
index f600612b..6cb6ffea 100644
--- a/demo/predict.py
+++ b/demo/predict.py
@@ -49,10 +49,17 @@ def get_parser_infer(parents=None):
 
     return parser
 
+def is_yolov7(args):
+    if "yolov7" not in args.config:
+        pass 
+    else:
+        ms.set_context(ascend_config={"precision_mode":"allow_fp32_to_fp16"})
 
 def set_default_infer(args):
     # Set Context
     ms.set_context(mode=args.ms_mode, device_target=args.device_target, max_call_depth=2000)
+    # MaxPool2d does not support dtype=fp32, ops's bug. Needed to be updated when ops's demand is done.
+    is_yolov7(args)
     if args.ms_mode == 0:
         ms.set_context(jit_config={"jit_level": "O2"})
     if args.device_target == "Ascend":
diff --git a/test.py b/test.py
index 3a6f0860..fe0a7b8e 100644
--- a/test.py
+++ b/test.py
@@ -67,10 +67,17 @@ def get_parser_test(parents=None):
     )
     return parser
 
+def is_yolov7(args):
+    if "yolov7" not in args.config:
+        pass 
+    else:
+        ms.set_context(ascend_config={"precision_mode":"allow_fp32_to_fp16"})
 
 def set_default_test(args):
     # Set Context
     ms.set_context(mode=args.ms_mode, device_target=args.device_target, max_call_depth=2000)
+    # MaxPool2d does not support dtype=fp32, ops's bug. Needed to be updated when ops's demand is done.
+    is_yolov7(args)
     if args.ms_mode == 0:
         ms.set_context(jit_config={"jit_level": "O2"})
     if args.device_target == "Ascend":