From 7dbf5f1244ca1209f7c8fcc86ee8756be65c8b37 Mon Sep 17 00:00:00 2001
From: tangwei12 <tangwei12@baidu.com>
Date: Tue, 21 Aug 2018 00:02:34 +0800
Subject: [PATCH] Update distribute_transpiler.py

---
 python/paddle/fluid/transpiler/distribute_transpiler.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py
index 5cc447f1dd8..5da6353e689 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
 
-- 
GitLab