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 e5be9686733e4801fb53ae16e73a64ba654c70a2..56adfa113323c8edca34b7af71021b3b963a684c 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")