From 38cf5531083054cd11b9627fa39ba7e4d6e09760 Mon Sep 17 00:00:00 2001
From: Xin Pan <panxin.grad@gmail.com>
Date: Thu, 25 Oct 2018 20:47:56 +0800
Subject: [PATCH] fix distributed

test=develop
---
 python/paddle/fluid/transpiler/distribute_transpiler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py
index 2192139f8d5..aed89c67e98 100644
--- a/python/paddle/fluid/transpiler/distribute_transpiler.py
+++ b/python/paddle/fluid/transpiler/distribute_transpiler.py
@@ -1717,8 +1717,8 @@ to transpile() call.")
         lr_ops = []
         block = self.origin_program.global_block()
         for op in block.ops:
-            if int(op.attr(RPC_OP_ROLE_ATTR_NAME)) == int(
-                    LR_SCHED_OP_ROLE_ATTR_VALUE):
+            if int(op.attr(RPC_OP_ROLE_ATTR_NAME)) | int(
+                    LR_SCHED_OP_ROLE_ATTR_VALUE) > 0:
                 lr_ops.append(op)
                 log("append lr op: ", op.type)
         return lr_ops
-- 
GitLab