diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py
index 5cc447f1dd8736fd88bc72890bd46ea383e3491a..5da6353e6892d05363e318a8a393d1b788631ae3 100644
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@@ -712,7 +712,7 @@ class DistributeTranspiler(object):
         slice_vars_and_attrs = []
         block_suffix = "block"
         for param in self.param_grad_ep_mapping[endpoint]["params"]:
-            orig_var_name, block_name, _ = self._get_varname_parts(param)
+            orig_var_name, block_name, _ = self._get_varname_parts(param.name)
             if not block_name:
                 continue