diff --git a/python/paddle/hapi/dynamic_flops.py b/python/paddle/hapi/dynamic_flops.py
index 382227ea8329778d767a47114496d9365e6def8c..9e2f78b559f186717aeb0c3629ff7e71fbc476c4 100644
--- a/python/paddle/hapi/dynamic_flops.py
+++ b/python/paddle/hapi/dynamic_flops.py
@@ -229,7 +229,7 @@ def dynamic_flops(model, inputs, custom_ops=None, print_detail=False):
         else:
             if m_type not in types_collection:
                 print(
-                    "Cannot find suitable count function for {}. Treat it as zero Macs.".
+                    "Cannot find suitable count function for {}. Treat it as zero FLOPs.".
                     format(m_type))
 
         if flops_fn is not None:
@@ -256,9 +256,9 @@ def dynamic_flops(model, inputs, custom_ops=None, print_detail=False):
             continue
         total_ops += m.total_ops
         total_params += m.total_params
-
-    total_ops = int(total_ops)
-    total_params = int(total_params)
+    if hasattr(m, 'total_ops') and hasattr(m, 'total_params'):
+        total_ops = int(total_ops)
+        total_params = int(total_params)
 
     if training:
         model.train()