fix distributed

test=develop

fix distributed
test=develop
38cf5531 · Xin Pan · d5d09672 · 38cf5531
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle/fluid/transpiler/distribute_transpiler.py python/paddle/fluid/transpiler/distribute_transpiler.py +2 -2

未找到文件。
--- 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