diff --git a/python/paddle/hapi/dynamic_flops.py b/python/paddle/hapi/dynamic_flops.py
index 8be6758f1e54b57adadd01025fe9c805556a67d7..2c59ee67d4a8e9ae49bf7eadec2d1b1e9d1e7fce 100644
--- a/python/paddle/hapi/dynamic_flops.py
+++ b/python/paddle/hapi/dynamic_flops.py
@@ -211,8 +211,8 @@ def dynamic_flops(model, inputs, custom_ops=None, print_detail=False):
     def add_hooks(m):
         if len(list(m.children())) > 0:
             return
-        m.register_buffer('total_ops', paddle.zeros([1], dtype='int32'))
-        m.register_buffer('total_params', paddle.zeros([1], dtype='int32'))
+        m.register_buffer('total_ops', paddle.zeros([1], dtype='int64'))
+        m.register_buffer('total_params', paddle.zeros([1], dtype='int64'))
         m_type = type(m)
 
         flops_fn = None