From a6ab43aab4501274f66cc8726bccce8cc5e46083 Mon Sep 17 00:00:00 2001
From: ShenLiang <2282912238@qq.com>
Date: Fri, 22 May 2020 19:57:25 +0800
Subject: [PATCH] add type for lsgd, test=develop (#24702)

---
 python/paddle/fluid/transpiler/collective.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/fluid/transpiler/collective.py b/python/paddle/fluid/transpiler/collective.py
index 42623337de..ae4befa004 100644
--- a/python/paddle/fluid/transpiler/collective.py
+++ b/python/paddle/fluid/transpiler/collective.py
@@ -314,7 +314,8 @@ class LocalSGD(Collective):
                     name=self.snapshot_name(param.name),
                     shape=param.shape,
                     persistable=True,
-                    stop_gradient=True)
+                    stop_gradient=True,
+                    dtype=param.dtype)
 
                 block._insert_op(
                     idx + 1,
-- 
GitLab