From fa0a8e92d5d3c592fea7deac7bb7dba64ac306d5 Mon Sep 17 00:00:00 2001
From: CLAY-panjw <1330286576@qq.com>
Date: Mon, 11 Jul 2022 13:56:35 +0800
Subject: [PATCH] v1 test

---
 .../npu_bridge/estimator/npu/npu_loss_scale_optimizer.py     | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tf_adapter/python/npu_bridge/estimator/npu/npu_loss_scale_optimizer.py b/tf_adapter/python/npu_bridge/estimator/npu/npu_loss_scale_optimizer.py
index e5be96867..56adfa113 100644
--- a/tf_adapter/python/npu_bridge/estimator/npu/npu_loss_scale_optimizer.py
+++ b/tf_adapter/python/npu_bridge/estimator/npu/npu_loss_scale_optimizer.py
@@ -66,6 +66,11 @@ class NPULossScaleOptimizer(lso.LossScaleOptimizer):
                                                                      cleared_float_status),
                                                             name="overflow_status_reduce_all")
             else:
+                for i in range(len(local_float_status)):
+                    print(local_float_status[i])
+                    print("************clear")
+                    print(cleared_float_status[i])
+                print(tf.equal(self._float_status, cleared_float_status))
                 is_overall_finite = math_ops.reduce_all(tf.equal(self._float_status,
                                                                  cleared_float_status),
                                                         name="overflow_status_reduce_all")
-- 
Gitee