Merge branch 'dev-2.0.0-beta-arch-update' of https://github.com/Feder…

…atedAI/FATE into feature-2.0.0-glm
FederatedAI · Jul 3, 2023 · 764e769 · 764e769
2 parents e5ad936 + dc59473
commit 764e769
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/fate/arch/tensor/distributed/_ops_binary.py b/python/fate/arch/tensor/distributed/_ops_binary.py
@@ -50,9 +50,10 @@ def _binary(input, other, op, swap_operad=False, dtype_promote_to=None):
         if swap_operad:
             return DTensor(
                 input.shardings.map_shard(
-                    lambda x: op(other, x, dtype_promote_to=dtype_promote_to), shapes=shapes.shapes, axis=shapes.axis
+                    lambda x: op(other, x), dtype_promote_to=dtype_promote_to, shapes=shapes.shapes, axis=shapes.axis
                 )
             )
+
         else:
             return DTensor(
                 input.shardings.map_shard(